Update main.py

main.py

@@ -56,7 +56,6 @@ try:
     pipe = pipeline(
         "text-generation",
         model=MODEL_CONFIG["model_name"],
-        device=device,
         max_new_tokens=MODEL_CONFIG["max_new_tokens"],
         temperature=0.3,
         do_sample=True, # Allow sampling to generate diverse responses. More conversational and human-like