Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def predict(message, history):
|
|
59 |
# break
|
60 |
# yield partial_message
|
61 |
inputs = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
|
62 |
-
outputs = model.generate(inputs, do_sample=False, max_length=
|
63 |
print(tokenizer.decode(outputs[0]))
|
64 |
return(tokenizer.decode(outputs[0]))
|
65 |
|
|
|
59 |
# break
|
60 |
# yield partial_message
|
61 |
inputs = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
|
62 |
+
outputs = model.generate(inputs, do_sample=False, max_length=500)
|
63 |
print(tokenizer.decode(outputs[0]))
|
64 |
return(tokenizer.decode(outputs[0]))
|
65 |
|