Update app.py

app.py

@@ -37,10 +37,10 @@ def text2story(text):
         str: Generated superhero story suitable for kids aged 3-10, within 100 words.
     """
     # Load the text generation model
-    story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")
+    story_generator = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
     
     # Generate the story with the superhero genre
-    prompt = f"<BOS> <superhero> {text}"
+    prompt = f"{text}"
     story = story_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
     
     # Remove <BOS> and <superhero> tags from the generated story
@@ -94,7 +94,7 @@ if uploaded_file is not None:
         file.write(bytes_data)
 
     # Display the uploaded image
-    st.image(uploaded_file, caption="Your awesome picture!", use_column_width=True)
+    st.image(uploaded_file, caption="Your awesome picture!", use_container_width=True)
 
     # Stage 1: Image to Text
     st.text('✨ Turning your picture into words...')