Update app.py
Browse files
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 |
-
|
182 |
-
|
183 |
-
|
184 |
-
|
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()
|