Update app.py

app.py

@@ -35,7 +35,9 @@ def generate(text, target_language):
         return_tensors="pt"
     )
 
-    outputs = model.generate(tokenized_chat.to(model.device), max_new_tokens=2048)
+    inputs = {k: v.to(model.device) for k, v in tokenized_chat.items()}
+
+    outputs = model.generate(inputs, max_new_tokens=2048)
 
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return result