Update pages/Captionize.py

pages/Captionize.py

@@ -33,8 +33,10 @@ div.stButton > button:hover {
 
 pic = st.file_uploader(label="Please upload any Image here 😎",type=['png', 'jpeg', 'jpg'], help="Only 'png', 'jpeg' or 'jpg' formats allowed")
 
+examples = [f"example{i}.jpg" for i in range(1,7)]
+
 #Image.open(requests.get(pic, stream=True).raw).convert("RGB")
-loader = ImageCaptionLoader(path_images=pic)
+loader = ImageCaptionLoader(path_images=examples)
 list_docs = loader.load()
 index = VectorstoreIndexCreator().from_loaders([loader])
 
@@ -42,7 +44,7 @@ button = st.button("Generate Caption")
 query = st.text_area("Enter your query 🔍")
 
 if button:
-    Image.open(requests.get(pic, stream=True).raw).convert("RGB")
+    Image.open(requests.get(examples[0], stream=True).raw).convert("RGB")
     # Get Response
     caption =  index.query(query)
     st.write(caption)