Update app.py

app.py

@@ -20,7 +20,7 @@ vae = get_vae().to(device)
 
 def translation_wrapper(text: str):
     input_ids = translation_tokenizer.encode(text, return_tensors="pt")
-    outputs = translation_model.generate(input_ids)
+    outputs = translation_model.generate(input_ids.to(device))
     decoded = translation_tokenizer.decode(outputs[0].float(), skip_special_tokens=True)
     return decoded