Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,8 +16,8 @@ st.set_page_config(page_title="Image2Story Generator", layout="centered")
 st.title("Image2Story Generator")
 st.write("Upload your Image and let us generate a Story for you")
 
-#create upload folder if not there
-os.makedirs("uploads", exist_ok=True)
+upload_dir = "/tmp/uploads"
+os.makedirs(upload_dir, exist_ok=True)
 
 #Upload your file
 uploaded_file = st.sidebar.file_uploader("Upload Your Image", type= ["jpg" , "jpeg" , "png" , "webp"])
@@ -26,7 +26,7 @@ uploaded_file = st.sidebar.file_uploader("Upload Your Image", type= ["jpg" , "jp
 
 if uploaded_file:
     image = Image.open(uploaded_file).convert("RGB")
-    image_path = os.path.join("uploads",uploaded_file.name)
+    image_path = oos.path.join(upload_dir, uploaded_file.name)
     image.save(image_path)
     
     # st.image(image, caption="uploaded image" , use_container_width=True)