Update app.py

app.py

@@ -1,12 +1,27 @@
+# import part
 import streamlit as st
 from PIL import Image
 import time
 
+from transformers import pipeline
+
+def generate_image_caption(image_path):
+    """Generates a caption for the given image using a pre-trained model."""
+    img2caption = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+    result = img2caption(image_path)
+    return result[0]['generated_text']
+
+# text2story
+def text2story(text):
+    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story_text = pipe(text)[0]['generated_text']
+    return story_text
+
 # App title
-st.title("Streamlit Demo on Hugging Face")
+st.title("Assignment")
 
 # Write some text
-st.write("Welcome to a demo app showcasing basic Streamlit components!")
+st.write("Image to Story")
 
 # File uploader for image and audio
 uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])