Update app.py

app.py

@@ -85,7 +85,7 @@ model = AutoModelForCausalLM.from_pretrained(conv_model_name, trust_remote_code=
 
     # output = pipe(text, **generation_args)
     # return output[0]['generated_text']
-    input_ids = tokenizer(text, return_tensors="pt").to(device)
+    input_ids = tokenizer(text, return_tensors="pt")
     outputs = model.generate(**input_ids)
     return tokenizer.decode(outputs[0])