Update app.py

app.py

@@ -41,7 +41,7 @@ pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
 
 # Translation logic
 def translate(text, target_language):
-    prompt = f"Translate the following text to {target_language}\n # Input:\n{text}\n\n# Output:\n"
+    prompt = f"Translate the following text to {target_language}\n # Input Text:\n{text}\n\n# Output Text:\n"
     ctx = prompt.strip()
     all_tokens = []
     out_last = 0
@@ -70,6 +70,117 @@ def translate(text, target_language):
     
     yield out_str.strip()
 
+# Languages
+LANGUAGES = [
+  "English",
+  "Chinese",
+  "Spanish",
+  "Bengali",
+  "Hindi", 
+  "Portuguese",
+  "Russian",
+  "Japanese",
+  "German",
+  "Chinese (Wu)",
+  "Javanese",
+  "Korean",
+  "French",
+  "Vietnamese",
+  "Telugu",
+  "Chinese (Yue)",
+  "Marathi", 
+  "Tamil",
+  "Turkish",
+  "Urdu",
+  "Chinese (Min Nan)",
+  "Chinese (Jin Yu)",
+  "Gujarati",
+  "Polish",
+  "Arabic (Egyptian Spoken)",
+  "Ukrainian",
+  "Italian",
+  "Chinese (Xiang)",
+  "Malayalam",
+  "Chinese (Hakka)",
+  "Kannada",
+  "Oriya",
+  "Panjabi (Western)",
+  "Panjabi (Eastern)",
+  "Sunda",
+  "Romanian",
+  "Bhojpuri",
+  "Azerbaijani (South)",
+  "Farsi (Western)",
+  "Maithili",
+  "Hausa",
+  "Arabic (Algerian Spoken)",
+  "Burmese",
+  "Serbo-Croatian",
+  "Chinese (Gan)",
+  "Awadhi",
+  "Thai",
+  "Dutch", 
+  "Yoruba",
+  "Sindhi",
+  "Arabic (Moroccan Spoken)",
+  "Arabic (Saidi Spoken)",
+  "Uzbek, Northern",
+  "Malay",
+  "Amharic",
+  "Indonesian",
+  "Igbo",
+  "Tagalog",
+  "Nepali",
+  "Arabic (Sudanese Spoken)",
+  "Saraiki",
+  "Cebuano",
+  "Arabic (North Levantine Spoken)",
+  "Thai (Northeastern)",
+  "Assamese",
+  "Hungarian",
+  "Chittagonian",
+  "Arabic (Mesopotamian Spoken)",
+  "Madura",
+  "Sinhala",
+  "Haryanvi",
+  "Marwari",
+  "Czech",
+  "Greek",
+  "Magahi",
+  "Chhattisgarhi",
+  "Deccan",
+  "Chinese (Min Bei)",
+  "Belarusan",
+  "Zhuang (Northern)",
+  "Arabic (Najdi Spoken)",
+  "Pashto (Northern)",
+  "Somali",
+  "Malagasy",
+  "Arabic (Tunisian Spoken)",
+  "Rwanda",
+  "Zulu",
+  "Bulgarian",
+  "Swedish",
+  "Lombard",
+  "Oromo (West-central)",
+  "Pashto (Southern)",
+  "Kazakh",
+  "Ilocano",
+  "Tatar",
+  "Fulfulde (Nigerian)",
+  "Arabic (Sanaani Spoken)",
+  "Uyghur",
+  "Haitian Creole French",
+  "Azerbaijani, North",
+  "Napoletano-calabrese",
+  "Khmer (Central)",
+  "Farsi (Eastern)",
+  "Akan",
+  "Hiligaynon",
+  "Kurmanji",
+  "Shona"
+]
+
 # Example data
 EXAMPLES = [
     ["Hello, how are you?", "French"],
@@ -89,8 +200,8 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown("This is the RWKV-5 World v2 1B5 model tailored for translation. Please provide the text and select the target language for translation.")
     
     # Input and output components
-    text = gr.Textbox(lines=5, label="English Text", placeholder="Enter the text you want to translate...")
-    target_language = gr.Dropdown(choices=["French", "Spanish", "German", "Chinese", "Japanese", "Russian", "Arabic"], label="Target Language")
+    text = gr.Textbox(lines=5, label="Source Text", placeholder="Enter the text you want to translate...")
+    target_language = gr.Dropdown(choices=LANGUAGES, label="Target Language")
     output = gr.Textbox(lines=5, label="Translated Text")
     submit = gr.Button("Translate", variant="primary")