Update app.py
Browse files
app.py
CHANGED
|
@@ -232,7 +232,7 @@ model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
|
|
| 232 |
def func (message):
|
| 233 |
inputs = tokenizer(message, return_tensors="pt")
|
| 234 |
result = model.generate(**inputs)
|
| 235 |
-
|
| 236 |
return tokenizer.decode(result[0])
|
| 237 |
|
| 238 |
|
|
|
|
| 232 |
def func (message):
|
| 233 |
inputs = tokenizer(message, return_tensors="pt")
|
| 234 |
result = model.generate(**inputs)
|
| 235 |
+
print(result)
|
| 236 |
return tokenizer.decode(result[0])
|
| 237 |
|
| 238 |
|