Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,7 @@ examples_correct = [
|
|
12 |
"Omnes legiones",
|
13 |
"reliqua omnia habet",
|
14 |
"difficiliores habent motus",
|
15 |
-
"nostra definitione retinetis"
|
16 |
-
"fossero guerre"
|
17 |
]
|
18 |
|
19 |
# Imposta una chiave di sessione per il testo input, così possiamo aggiornarlo
|
@@ -64,8 +63,7 @@ if input_text:
|
|
64 |
st.write(f" -----------------------------------------------------------\n")
|
65 |
|
66 |
inputs_mio = tokenizer_mio(input_text, return_tensors="pt")
|
67 |
-
output_mio =
|
68 |
-
**inputs_mio,
|
69 |
#num_return_sequences=1,
|
70 |
#top_k=50, # 🔹 Maggiore varietà nelle scelte
|
71 |
#top_p=0.95, # 🔹 Nucleus sampling per migliorare il realismo
|
@@ -73,7 +71,7 @@ if input_text:
|
|
73 |
#repetition_penalty=1.2, # 🔹 Evita ripetizioni e loop
|
74 |
#max_length=50 # 🔹 Previene ripetizioni infinite
|
75 |
)
|
76 |
-
generated_text_mio = tokenizer_mio.decode(output_mio[0], skip_special_tokens=True)
|
77 |
-
|
78 |
st.subheader("Risultato BART CC100:")
|
79 |
st.write(f" Frase predetta: {generated_text_mio}\n")
|
|
|
12 |
"Omnes legiones",
|
13 |
"reliqua omnia habet",
|
14 |
"difficiliores habent motus",
|
15 |
+
"nostra definitione retinetis"
|
|
|
16 |
]
|
17 |
|
18 |
# Imposta una chiave di sessione per il testo input, così possiamo aggiornarlo
|
|
|
63 |
st.write(f" -----------------------------------------------------------\n")
|
64 |
|
65 |
inputs_mio = tokenizer_mio(input_text, return_tensors="pt")
|
66 |
+
output_mio = generator_mio(input_text
|
|
|
67 |
#num_return_sequences=1,
|
68 |
#top_k=50, # 🔹 Maggiore varietà nelle scelte
|
69 |
#top_p=0.95, # 🔹 Nucleus sampling per migliorare il realismo
|
|
|
71 |
#repetition_penalty=1.2, # 🔹 Evita ripetizioni e loop
|
72 |
#max_length=50 # 🔹 Previene ripetizioni infinite
|
73 |
)
|
74 |
+
#generated_text_mio = tokenizer_mio.decode(output_mio[0], skip_special_tokens=True)
|
75 |
+
generated_text_mio = output_mio[0]["generated_text"]
|
76 |
st.subheader("Risultato BART CC100:")
|
77 |
st.write(f" Frase predetta: {generated_text_mio}\n")
|