Update app.py

app.py

@@ -1,18 +1,20 @@
 import gradio as gr
 from transformers import pipeline
+from langdetect import detect
 
 def translate_transformers(text, target_language):
-    translator = pipeline("translation", model=f"Helsinki-NLP/opus-mt-autodetect-{target_language}")
+    source_language = detect(text)
+    translator = pipeline("translation", model=f"Helsinki-NLP/opus-mt-{source_language}-{target_language}")
     translation = translator(text, max_length=4000)
     return translation[0]['translation_text']
 
 interface = gr.Interface(
     fn=translate_transformers,
     inputs=[
-        gr.components.Textbox(label="Text to translate"),
-        gr.components.Dropdown(["en", "fr", "de" , "es"], label="Target Language")
+        gr.inputs.Textbox(label="Text to translate"),
+        gr.inputs.Dropdown(["en", "fr", "de", "es", "ar", "zh-cn", "zh-tw", "nl", "hi", "it", "ja", "ko", "pt", "ru", "tr"], label="Target Language")
     ],
-    outputs=gr.components.Textbox(label="Translation")
+    outputs=gr.outputs.Textbox(label="Translation")
 )
 
 interface.launch()