Update app.py

app.py

@@ -16,7 +16,7 @@ question_answering = pipeline("question-answering", model="deepset/roberta-base-
 code_generation = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=device)
 summarization = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
 #translation = pipeline("translation_en_to_fr", model="Helsinki-NLP/opus-mt-en-fr", device=device)
-translation = pipeline("translation", model="facebook/m2m100_418M", device=device)
+#translation = pipeline("translation", model="facebook/m2m100_418M", device=device)
 text_generation = pipeline("text-generation", model="gpt2", device=device)
 text_classification = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english", device=device)
 
@@ -36,10 +36,7 @@ def summarize_text(text):
     output = summarization(text, max_length=100, min_length=30, do_sample=False)
     return output[0]['summary_text']
 
-@spaces.GPU
-def translate_text(text):
-    output = translation(text)
-    return output[0]['translation_text']
+
 
 @spaces.GPU
 def generate_text(prompt):
@@ -75,12 +72,6 @@ with gr.Blocks() as demo:
         summary_output = gr.Textbox(label="Summary")
         summary_btn.click(summarize_text, inputs=summary_input, outputs=summary_output)
 
-    with gr.Tab("4️⃣ Translation (EN → FR)"):
-        translate_input = gr.Textbox(label="English Text", placeholder="Enter text in English...")
-        translate_btn = gr.Button("Translate")
-        translate_output = gr.Textbox(label="French Translation")
-        translate_btn.click(translate_text, inputs=translate_input, outputs=translate_output)
-
     with gr.Tab("5️⃣ Text Generation"):
         text_input = gr.Textbox(label="Text Prompt", placeholder="Start your text...")
         text_btn = gr.Button("Generate Text")