Update app.py

app.py

@@ -10,18 +10,17 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    messages = f"""Generate a story of about 100 words using the following sentences: {text}."""
+    #messages = f"""Generate a story of about 100 words using the following sentences: {text}."""
    # messages = [
     #    {"role": "system", "content": "You're a chatbot that writes stories based on keywords.The story must contain:Beginning, middle, end，Character development"},
     #    {"role": "user", "content": "Keywords:"+text},
     #]
-    text_to_story_model= pipeline("text-generation", model="Alexis-Az/Story-Generation-Model")
+    text_to_story_model= pipeline("text-generation", model="Tincando/fiction_story_generator")
     story_text = text_to_story_model(
-        messages,
+        text,
         max_length=150,  
         min_length=50,
-        do_sample=True,
-        #early_stopping=False
+        #do_sample=True,
     )[0]['generated_text']
     return story_text