Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+from transformers import pipeline
 
 def img2text(url):
     image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
@@ -7,7 +7,7 @@ def img2text(url):
     return text
 
 def text2story(text):
-    story_generator = pipeline("text2text-generation", model="google/flan-t5-xl")
+    story_generator = pipeline("text-generation", model="togethercomputer/RedPajama-INCITE-7B-Instruct", device_map="auto")
     prompt = f"Create a short story under 100 words based on: {text}"
     generated = story_generator(prompt)
     story_text = generated[0]['generated_text']