Update app.py

app.py

@@ -3,6 +3,25 @@ from transformers import BartTokenizer, BartForConditionalGeneration
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelWithLMHead, TranslationPipeline
 
+
+
+from transformers import pipeline
+import gradio as gr
+pipe= pipeline('text2text-generation', model="SEBIS/legal_t5_small_trans_fr_en")
+def generate_text(inp):
+    output=pipe(inp, forced_bos_token_id=pipe.tokenizer.get_lang_id('en'))
+    tln=output[0]
+    for item in tln:
+        result=tln[item]
+    return result
+#Gradio Interface
+output_text = gr.outputs.Textbox()
+gr.Interface(generate_text,"textbox", output_text).launch(inline=False)
+
+
+
+
+
 model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
 tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")