Update app.py

app.py

@@ -17,9 +17,7 @@ except ImportError:
 # Load the image captioning model
 caption_model = pipeline("image-to-text", model="unography/blip-large-long-cap")
 
-#story_generator = pipeline("text-generation", model="distilbert/distilgpt2")
-
-story_generator = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
+story_generator = pipeline("text-generation", model="distilbert/distilgpt2")
 
 
 def generate_caption(image):