Try a different prediction method

app.py

@@ -38,18 +38,15 @@ def formal_to_dostoevsky(texto_informal):
     )
     
     # Forzamos parámetros de generación específicos para transformación de estilo
-    output_ids = model.generate(
-        inputs["input_ids"],
-        attention_mask=inputs["attention_mask"],
-        max_length=300,
-        min_length=30,       # Forzamos a que desarrolle la idea
-        num_beams=5,
-        do_sample=True,      # Necesario para usar temperature
-        temperature=0.75,
-        top_p=0.80,
-        no_repeat_ngram_size=3,
-        repetition_penalty=1.7,
-        early_stopping=True
+    outputs = model.generate(
+        input_ids,
+        max_length=100,
+        do_sample=True,         # Permite creatividad
+        top_k=50,               # Filtra palabras raras
+        top_p=0.95,             # Nucleus sampling para coherencia
+        temperature=0.9,        # Evita que se vuelva loco o muy plano
+        repetition_penalty=2.5, # ¡Clave! Evita que repita "bread" o "Today"
+        no_repeat_ngram_size=3  # Evita que repita frases de 3 palabras
     )
     
     # Decodificar omitiendo los tokens especiales