Upload app.py

app.py

@@ -29,7 +29,7 @@ examples = [
 
 @measure_performance
 def generate_story(theme):
-    prompt = [{"role": "user", "content": f"Tell me a storie about {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)
     generated_story = story[0]["generated_text"]
     return generated_story