Update app.py

app.py

@@ -9,7 +9,7 @@ checkpoint = "rasyosef/Phi-1_5-Instruct-v0.1"
 
 # Download and load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float32, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float32, device_map="cpu")
 
 # Text generation pipeline
 phi1_5 = pipeline(