Upload app.py

app.py

@@ -27,6 +27,9 @@ def measure_performance(func):
 
 @measure_performance
 def generate_story_text(theme):
+    if generate_story_pipe.tokenizer.pad_token is None:
+        generate_story_pipe.tokenizer.pad_token = generate_story_pipe.tokenizer.eos_token
+
     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, truncation=True, padding=True)
     generated_story = story[0]["generated_text"]