Hugging Face's logo

Spaces:
Jumpy-pku
/
dish-name-recognition
Sleeping

App Files Community
Jumpy-pku commited on
Commit
64d41ef
•
1 Parent(s): 09af246

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -1,10 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # pipe = pipeline(task="text2text-generation", model="Jumpy-pku/t5-dish-name-recognition")
5
- # pipe_aux = pipeline(task="text2text-generation", model="Jumpy-pku/t5-dish-name-recognition-auxiliary")
6
- pipe = pipeline(task="text2text-generation", model="model/t5-name-cot")
7
- pipe_aux = pipeline(task="text2text-generation", model="model/t5-component")
8
 
9
  def predict(input_text):
10
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline(task="text2text-generation", model="Jumpy-pku/t5-dish-name-recognition")
5
+ pipe_aux = pipeline(task="text2text-generation", model="Jumpy-pku/t5-dish-name-recognition-auxiliary")
 
 
6
 
7
  def predict(input_text):
8