Update app.py

app.py

@@ -7,8 +7,8 @@ def img2text(url):
     return text
 
 def text2story(text):
-    story_generator = pipeline("text-generation", model="openai-community/gpt2")
-    prompt = f"Create a short story under 100 words based on: {text}"
+    story_generator = pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)
+    prompt = [{"role": "user", "content": f"Create a short story under 100 words based on: {text}"}]
     generated = story_generator(prompt, max_length=100)
     story_text = generated[0]['generated_text']
     return story_text