Update app.py

app.py

@@ -27,21 +27,22 @@ def load_model():
     """Loads the Mistral model and tokenizer and updates the load status."""
     global tokenizer, model, model_loaded
     try:
-        if tokenizer is None:
-            tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
-        if model is None:
-            model = AutoModelForCausalLM.from_pretrained(
-                model_name,
-                token=hf_token,
-                torch_dtype=torch.float16
-            )
-            model.to('cuda')
+        if model is not None and tokenizer is not None:  # Prevent unnecessary reloads
+            model_loaded = True
+            return "✅ Model Already Loaded!"
+
+        tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
+        model = AutoModelForCausalLM.from_pretrained(
+            model_name,
+            token=hf_token,
+            torch_dtype=torch.float16
+        )
+        model.to('cuda')
         model_loaded = True
         return "✅ Model Loaded!"
     except Exception as e:
         model_loaded = False
         return f"❌ Model Load Failed: {str(e)}"
-
 # ---------------------------------------------------------------------------
 # 2. Toy Evaluation
 # ---------------------------------------------------------------------------