Chris4K commited on
Commit
b8d0c43
1 Parent(s): d45af61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -171,20 +171,20 @@ model_name = 'facebook/blenderbot-400M-distill'
171
  tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
172
  model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
173
 
174
- def func (message):
175
  inputs = tokenizer(message, return_tensors="pt")
176
  result = model.generate(**inputs)
177
- return tokenizer.decode(result[0])
178
 
179
 
180
 
181
- #app = gr.Interface(
182
- # fn=func,
183
- # inputs=["text", "checkbox", gr.Slider(0, 100)],
184
- # outputs=["text", "number"],
185
- #)
186
 
187
 
188
- app = gr.Interface(fn=func , inputs="textbox", outputs="textbox", title="Conversation Bot")
189
  # create a public link, set `share=True` in `launch()
190
  app.launch()
 
171
  tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
172
  model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
173
 
174
+ def func (message,text, numb):
175
  inputs = tokenizer(message, return_tensors="pt")
176
  result = model.generate(**inputs)
177
+ return tokenizer.decode(result[0]),9
178
 
179
 
180
 
181
+ app = gr.Interface(
182
+ fn=func,
183
+ inputs=["text", "checkbox", gr.Slider(0, 100)],
184
+ outputs=["text", "number"],
185
+ )
186
 
187
 
188
+ #app = gr.Interface(fn=func , inputs="textbox", outputs="textbox", title="Conversation Bot")
189
  # create a public link, set `share=True` in `launch()
190
  app.launch()