Update app.py

app.py

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