Update app.py

app.py

@@ -15,14 +15,14 @@ def translate_text(text, translation_direction):
         inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
         outputs = model.generate(**inputs)
         translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return json.dumps({"input_text": text, "translated_text": translated_text, "language": "en-to-iban"})
+        return json.dumps({"input_text": text, "translated_text": translated_text, "language": "en-to-iba"})
     
     elif translation_direction == "Iban to English":
-        # Translate Iban to English (you'll need a reverse model or translation setup here)
+        # Translate Iban to English
         inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
         outputs = model.generate(**inputs)
         translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return json.dumps({"input_text": text, "translated_text": translated_text, "language": "iban-to-en"})
+        return json.dumps({"input_text": text, "translated_text": translated_text, "language": "iba-to-en"})
 
 # Define the Gradio interface with a dropdown to select translation direction
 iface = gr.Interface(