Spaces:
Runtime error
Runtime error
Update app.py
Browse filesFix a bug in the infer function
app.py
CHANGED
@@ -12,7 +12,7 @@ model = load_model()
|
|
12 |
|
13 |
def infer(input_ids):
|
14 |
output_sequences = model.generate(input_ids=input_ids)
|
15 |
-
return tokenizer.decode(
|
16 |
|
17 |
st.title("Stupid jokes with transformers")
|
18 |
st.write("Write question you want to see a funny answer for.")
|
|
|
12 |
|
13 |
def infer(input_ids):
|
14 |
output_sequences = model.generate(input_ids=input_ids)
|
15 |
+
return tokenizer.decode(output_sequences[0], skip_special_tokens=True)
|
16 |
|
17 |
st.title("Stupid jokes with transformers")
|
18 |
st.write("Write question you want to see a funny answer for.")
|