Update app.py

app.py

@@ -2,7 +2,11 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/deepseek-coder-6.7b-instruct", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("deepseek-ai/deepseek-coder-6.7b-instruct", trust_remote_code=True, device_map="auto")
+model = AutoModelForCausalLM.from_pretrained(
+    "deepseek-ai/deepseek-coder-6.7b-instruct", 
+    trust_remote_code=True
+)
+
 
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)