Update app.py

app.py

@@ -11,15 +11,19 @@ import json
 MODEL_NAME = "Vilyam888/Code_analyze.1.0"
 
 def load_model():
-    """Загружает модель один раз при старте"""
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
-    model = AutoModelForCausalLM.from_pretrained(
-        MODEL_NAME,
-        torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
-        device_map="auto",
-        trust_remote_code=True
-    )
-    return model, tokenizer
+    """Загружает модель один раз при старте и сохраняет в глобальные переменные"""
+    global model, tokenizer
+    
+    if model is None or tokenizer is None:
+        print("Загрузка модели...")
+        tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)
+        model = AutoModelForCausalLM.from_pretrained(
+            MODEL_NAME,
+            torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
+            device_map="auto",
+            trust_remote_code=True
+        )
+        print("Модель загружена успешно!")
 
 # Глобальные переменные для модели
 model, tokenizer = None, None