Update app.py

app.py

@@ -27,17 +27,15 @@ def load_model():
     """Loads the Mistral model and tokenizer and updates the load status."""
     global tokenizer, model, model_loaded
     try:
-        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')
+        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')
         model_loaded = True
         return "✅ Model Loaded!"
     except Exception as e: