gokilashree commited on
Commit
6f13ba7
·
verified ·
1 Parent(s): 6804466

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -6,6 +6,16 @@ import io
6
  from PIL import Image
7
 
8
  !pip install torch --upgrade
 
 
 
 
 
 
 
 
 
 
9
  … fn=translate_and_generate_image,
10
  inputs=gr.Textbox(lines=2, placeholder="Enter Tamil text here..."),
11
  outputs=[gr.Textbox(label="Translated English Text"), gr.Image(label="Generated Image")],
 
6
  from PIL import Image
7
 
8
  !pip install torch --upgrade
9
+ !pip install transformers
10
+ !pip install sentencepiece
11
+ !pip install huggingface_hub
12
+ !pip install gradio --upgrade
13
+ !pip install diffusers
14
+ !pip install pillow
15
+
16
+
17
+ # Load Facebook's mbart-large-50 model and tokenizer
18
+ model_name = "facebook/mbart-large-50-many-to-one-mmt"
19
  … fn=translate_and_generate_image,
20
  inputs=gr.Textbox(lines=2, placeholder="Enter Tamil text here..."),
21
  outputs=[gr.Textbox(label="Translated English Text"), gr.Image(label="Generated Image")],