Update app.py

app.py

@@ -28,7 +28,7 @@ model_name = "mistralai/Mistral-7B-Instruct-v0.3"  # fictional placeholder
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token, torch_dtype=torch.float16, device_map="auto").to(device)
+model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token, torch_dtype=torch.float16, device_map="auto")
 
 print(f"✅ Model loaded on {device}")
 #model = AutoModelForCausalLM.from_pretrained(model_name)