Update app.py

app.py

@@ -95,7 +95,7 @@ def generate_story(caption: str) -> str:
     """
     try:
         cap2story = load_story_pipeline()
-        output = cap2story(f"<BOS> <adventure>{text}",truncation=True,max_length=80,do_sample=True,repetition_penalty=1.5, temperature=1.2,top_p=0.95,top_k=50,no_repeat_ngram_size=2)
+        output = cap2story(f"<BOS> <adventure>{caption}",truncation=True,max_length=80,do_sample=True,repetition_penalty=1.5, temperature=1.2,top_p=0.95,top_k=50,no_repeat_ngram_size=2)
         story = output[0]['generated_text'].replace("<BOS> <adventure>","")
         return story
     except Exception as e: