Update app.py

app.py

@@ -31,7 +31,8 @@ def generate(prompt, history):
         conversation,
         add_generation_prompt=True,
         tokenize=True,
-        return_tensors="pt"
+        return_tensors="pt",
+        return_dict=True
     ).to(device)
     
     with torch.no_grad():