Fried commited on
Commit
d7af86b
·
1 Parent(s): a6be5ad

Add Gradio interface for translation using transformers

Browse files
Files changed (3) hide show
  1. app.py +5 -2
  2. pyproject.toml +1 -0
  3. transformers_app.py +7 -0
app.py CHANGED
@@ -25,5 +25,8 @@ demo = gr.Interface(
25
  outputs=gr.Label(num_top_classes=3),
26
  examples=["lion.jpg", "cheetah.jpg"])
27
 
28
- if __name__ == "__main__":
29
- demo.launch(show_api=True, share=True)
 
 
 
 
25
  outputs=gr.Label(num_top_classes=3),
26
  examples=["lion.jpg", "cheetah.jpg"])
27
 
28
+ # if __name__ == "__main__":
29
+
30
+ demo.launch(
31
+ # show_api=True, share=True
32
+ )
pyproject.toml CHANGED
@@ -13,6 +13,7 @@ requests = "^2.31.0"
13
  pillow = "^10.1.0"
14
  torchvision = "^0.16.2"
15
  matplotlib = "^3.8.2"
 
16
 
17
 
18
  [build-system]
 
13
  pillow = "^10.1.0"
14
  torchvision = "^0.16.2"
15
  matplotlib = "^3.8.2"
16
+ transformers = {extras = ["sentencepiece"], version = "^4.36.2"}
17
 
18
 
19
  [build-system]
transformers_app.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ import gradio as gr
3
+
4
+ pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
5
+
6
+ demo = gr.Interface.from_pipeline(pipe)
7
+ demo.launch()