Update app.py

app.py

@@ -1,28 +1,21 @@
 import gradio as gr
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
 # Loading translation model
 translation_tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_translation")
 translation_model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_translation")
-translate = pipeline("text2text-generation", model=translation_model, tokenizer=translation_tokenizer)
+translator = pipeline("translation_en_to_fr", tokenizer=translation_tokenizer, model=translation_model)
 
-# Loading summarization model
-summarization_tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_summarization")
-summarization_model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_summarization")
-summarize = pipeline("text2text-generation", model=summarization_model, tokenizer=summarization_tokenizer)
+def translate(text):
+    return translator(text)[0]['translation_text']
 
-iface = gr.Interface(
-    fn=gr.mix.Parallel(summarize, translate),
-    inputs=[
-        gr.inputs.Textbox(lines=10, label="Input Text"),
-        gr.inputs.Dropdown(choices=["summarize", "translate"], label="Task")
-    ],
-    outputs=[
-        gr.outputs.Textbox(label="Summary"),
-        gr.outputs.Textbox(label="Translation")
-    ],
+iface_translate = gr.Interface(
+    fn=translate,
+    inputs=gr.inputs.Textbox(lines=10, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Translation"),
+    title="ScienceBrief Translation",
+    description="Translate your text into French using the ScienceBrief translation model.",
+    theme="compact"
 )
 
-iface.launch()
-
-
+iface_translate.launch()