Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
-import torch
+from transformers import pipeline
+
 # function part
 # img2text
 def img2text(url):
@@ -10,7 +10,7 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    text_to_story = pipeline("text-generation", model="TheBloke/storytime-13B-GPTQ")
+    text_to_story = pipeline("text-generation", model="ajibawa-2023/Young-Children-Storyteller-Mistral-7B")
     # story_text = ""   # to be completed
     story_text = text_to_story(text)[0]["generated_text"]
     return story_text