Update app.py

app.py

@@ -27,10 +27,9 @@ def mistral_model():
     Returns:
         tuple: A tuple containing the loaded model and tokenizer.
     """
-    if torch.cuda.is_available():
-        model_id = "/code/mistral/"
-        model = AutoModelForCausalLM.from_pretrained("/home/user/app/mistral/")
-        tokenizer = AutoTokenizer.from_pretrained("/home/user/app/mistral/")
+    model_id = "/code/mistral/"
+    model = AutoModelForCausalLM.from_pretrained("/home/user/app/mistral/")
+    tokenizer = AutoTokenizer.from_pretrained("/home/user/app/mistral/")
 
     return model,tokenizer