Update app.py
Browse files
app.py
CHANGED
|
@@ -171,15 +171,16 @@ 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]),9
|
| 178 |
|
| 179 |
|
| 180 |
|
| 181 |
app = gr.Interface(
|
| 182 |
fn=func,
|
|
|
|
| 183 |
inputs=["text", "checkbox", gr.Slider(0, 100)],
|
| 184 |
outputs=["text", "number"],
|
| 185 |
)
|
|
|
|
| 171 |
tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
|
| 172 |
model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
|
| 173 |
|
| 174 |
+
def func (message, checkbox, numb):
|
| 175 |
inputs = tokenizer(message, return_tensors="pt")
|
| 176 |
result = model.generate(**inputs)
|
| 177 |
+
return tokenizer.decode(result[0]),"0.9"
|
| 178 |
|
| 179 |
|
| 180 |
|
| 181 |
app = gr.Interface(
|
| 182 |
fn=func,
|
| 183 |
+
title="Conversation Bota"
|
| 184 |
inputs=["text", "checkbox", gr.Slider(0, 100)],
|
| 185 |
outputs=["text", "number"],
|
| 186 |
)
|