Update app.py

app.py

@@ -4,7 +4,10 @@ import torch
 
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", device="cpu", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", trust_remote_code=True)
+
+# Move model to CPU
+model.to("cpu")
 
 # Streamlit app
 st.title("Text Generation with Transformers")