Spaces:
Runtime error
Runtime error
Commit ·
6d06124
1
Parent(s): 6c03bb7
Update app.py
Browse files
app.py
CHANGED
|
@@ -81,7 +81,6 @@ def generate_title():
|
|
| 81 |
# compute predictions
|
| 82 |
outputs = model.generate(**inputs, do_sample=True, temperature=temperature, max_new_tokens=64)
|
| 83 |
decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
| 84 |
-
predicted_titles = decoded_output
|
| 85 |
|
| 86 |
st.session_state.titles = decoded_outputs
|
| 87 |
|
|
|
|
| 81 |
# compute predictions
|
| 82 |
outputs = model.generate(**inputs, do_sample=True, temperature=temperature, max_new_tokens=64)
|
| 83 |
decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
|
|
|
|
| 84 |
|
| 85 |
st.session_state.titles = decoded_outputs
|
| 86 |
|