Update app.py

app.py

@@ -25,7 +25,7 @@ def generate(text):
        chat_data.append({"role": "system", "content": text})
     achat=tokenizer.apply_chat_template(chat_data,return_tensors="pt")
     #achat=tokenizer.encode(chat_data,return_tensors="pt",max_length=2048)
-    outputs = model.generate(achat.to('cuda'),
+    outputs = model.generate(achat,
                          max_new_tokens=2048,
                          top_p=0.01,
                          top_k=85,