Update app.py

app.py

@@ -16,7 +16,8 @@ torch.set_default_device(device)
 tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/phi-2",
-    torch_dtype="auto",
+    # torch_dtype=torch.float16 if has_gpu else torch.float32,
+    torch_dtype=torch.float32,
     device_map=device,
     trust_remote_code=True,
 )