Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -74,7 +74,7 @@ if input_text:
|
|
74 |
#repetition_penalty=1.2, # 🔹 Evita ripetizioni e loop
|
75 |
#max_length=50 # 🔹 Previene ripetizioni infinite
|
76 |
)
|
77 |
-
generated_text_mio = tokenizer_mio.
|
78 |
#generated_text_mio = output_mio[0]["generated_text"]
|
79 |
st.subheader("Risultato BART CC100:")
|
80 |
st.write(f" Frase predetta: {generated_text_mio}\n")
|
|
|
74 |
#repetition_penalty=1.2, # 🔹 Evita ripetizioni e loop
|
75 |
#max_length=50 # 🔹 Previene ripetizioni infinite
|
76 |
)
|
77 |
+
generated_text_mio = tokenizer_mio.decode(output_mio[0], skip_special_tokens=True)
|
78 |
#generated_text_mio = output_mio[0]["generated_text"]
|
79 |
st.subheader("Risultato BART CC100:")
|
80 |
st.write(f" Frase predetta: {generated_text_mio}\n")
|