Cicciokr commited on
Commit
ed8f158
·
verified ·
1 Parent(s): 7e0aa86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
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 = model_mio.generate(
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
- #generated_text_mio = output_mio[0]["generated_text"]
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")