Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, lengt
|
|
30 |
eos_token_id=tokenizer.eos_token_id
|
31 |
)
|
32 |
|
33 |
-
return
|
34 |
|
35 |
|
36 |
model, tokenizer = get_model('facebook/opt-13b')
|
|
|
30 |
eos_token_id=tokenizer.eos_token_id
|
31 |
)
|
32 |
|
33 |
+
return list(map(tokenizer.decode, out))[0]
|
34 |
|
35 |
|
36 |
model, tokenizer = get_model('facebook/opt-13b')
|