Khadidja22 commited on
Commit
b5f9198
1 Parent(s): e5b5ba1

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
+
4
+ model_name = "my_awesome_opus_books_model"
5
+
6
+ # Load the tokenizer and model
7
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
8
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
9
+
10
+ def translate(text):
11
+ inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512).input_ids
12
+ outputs = model.generate(inputs, max_length=512, num_beams=5, early_stopping=True)
13
+ translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
14
+ return translated_text
15
+
16
+ # Create a Gradio interface
17
+ interface = gr.Interface(fn=translate,
18
+ inputs=gr.Textbox(lines=2, placeholder="Enter English text to translate..."),
19
+ outputs=gr.Textbox(),
20
+ title="English to French Translation",
21
+ description="Translate English text to French using a fine-tuned model, start your sentence with translate English to French:")
22
+
23
+ # Launch the interface
24
+ interface.launch()