upd

app.py

@@ -9,7 +9,7 @@ def initialize_model():
     model_config = HIMConfig()
     env_config = EnvironmentConfig()
     device = torch.device("cuda" if torch.cuda.is_available() and env_config.device == "cuda" else "cpu")
-    model = HIMModel(model_config).to(device)  # Pass model_config directly instead of model_config.to_dict()
+    model = HIMModel(model_config.to_dict()).to(device)
     return model
 
 async def chat(message: str,

config/environment_config.py

@@ -1,10 +1,6 @@
-from dataclasses import dataclass
-from typing import Optional
-
-@dataclass
 class EnvironmentConfig:
     def __init__(self):
-        self.device = "cuda"  # or "cpu"
+        self.device = "cuda"  # ou "cpu"
         self.log_level = "INFO"
         self.api_port = 8000
         self.threads = 4

config/model_config.py

@@ -1,7 +1,3 @@
-from dataclasses import dataclass
-from typing import Dict, List
-
-@dataclass
 class HIMConfig:
     def __init__(self):
         self.base_model = "gpt2-medium"