Upload app.py

app.py

@@ -28,7 +28,7 @@ def measure_performance(func):
 @measure_performance
 def generate_story_text(theme):
     prompt = f"Tell me a storie about {theme}"
-    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)
+    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, truncation=True, padding=True)
     generated_story = story[0]["generated_text"]
     return generated_story