Update app.py

app.py

@@ -12,9 +12,15 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    text_to_story_model = pipeline("text-generation",model="distilbert/distilgpt2")
-    story_text = text_to_story_model(text,max_length=100,num_return_sequences=1)
-    return story_text
+    text_to_story_model = pipeline("text-generation", model="distilbert/distilgpt2") 
+    if isinstance(text, list):  
+        text = " ".join(text)  # Ensure input is a single string
+    story_text = text_to_story_model(text, max_length=100, num_return_sequences=1)
+    return story_text[0]['generated_text'] 
+    
+    #text_to_story_model = pipeline("text-generation",model="distilbert/distilgpt2")
+    #story_text = text_to_story_model(text,max_length=100,num_return_sequences=1)
+    #return story_text
 
 # text2audio
 def text2audio(story_text):