Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 # Load your model and tokenizer (same as before)
 model = AutoModelForCausalLM.from_pretrained(
     "marketeam/PhiMarketing",
-    device_map="cuda",
+    device_map="auto",
     torch_dtype="auto",
     trust_remote_code=True,
 )