Update app.py

app.py

@@ -1,58 +1,3 @@
 import gradio as gr
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-# Load the NLLB-200 model and tokenizer
-model_name = "facebook/nllb-200-distilled-600M"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
-# Translation function
-def translate(text, src_lang, tgt_lang):
-    # Convert the human-readable language names to their language codes
-    src_lang_code = lang_dict[src_lang]
-    tgt_lang_code = lang_dict[tgt_lang]
-    
-    # Tokenize input text with the source language code
-    inputs = tokenizer(text, return_tensors="pt", src_lang=src_lang_code)
-    
-    # Generate translation with the target language code
-    outputs = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[tgt_lang_code])
-    
-    # Decode and return the translation
-    translation = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return translation
-
-# List of languages supported by the model with their NLLB-200 codes
-lang_dict = {
-    "English": "eng_Latn",
-    "Spanish": "spa_Latn",
-    "French": "fra_Latn",
-    "German": "deu_Latn",
-    "Chinese": "zho_Hans",
-    "Hindi": "hin_Deva",
-    "Indonesian": "ind_Latn",   # Indonesian
-    "Malaysian": "zsm_Latn",    # Malaysian
-    "Javanese": "jav_Latn",     # Javanese
-    "Sundanese": "sun_Latn",    # Sundanese
-    # Add more languages if needed
-}
-
-# Define the Gradio interface using Blocks
-with gr.Blocks() as ui:
-    gr.Markdown("# Language Translation with NLLB-200 Model")
-    
-    with gr.Row():
-        src_lang = gr.Dropdown(label="Source Language", choices=list(lang_dict.keys()), value="English")
-        tgt_lang = gr.Dropdown(label="Target Language", choices=list(lang_dict.keys()), value="Indonesian")
-    
-    text_input = gr.Textbox(label="Enter text to translate", placeholder="Type your text here...")
-    translate_button = gr.Button("Translate")
-    translation_output = gr.Textbox(label="Translation Output")
-    
-    # Trigger translation on button click
-    translate_button.click(fn=translate, 
-                           inputs=[text_input, src_lang, tgt_lang], 
-                           outputs=translation_output)
-
-# Launch the UI
-ui.launch()
+gr.load("models/facebook/nllb-200-distilled-600M").launch()