Upload app.py

app.py

@@ -29,7 +29,7 @@ examples = [
 @measure_performance
 def generate_story(theme):
     prompt = f"Crea una historia {theme} en español"
-    story = generate_story_pipe(prompt, max_length=400, num_return_sequences=1, temperature=0.7, do_sample=True, repetition_penalty=3.0)
+    story = generate_story_pipe(prompt, max_length=400, num_return_sequences=1, temperature=0.7, do_sample=True, repetition_penalty=3.0, eos_token_id=generate_story_pipe.model.config.eos_token_id, pad_token_id=generate_story_pipe.model.config.pad_token_id)
     generated_story = story[0]["generated_text"]
     return generated_story