Update app.py

app.py

@@ -43,15 +43,17 @@ def main():
     # create file uploader
     uploaded_file  = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
-    # load the image
-    img = Image.open(uploaded_file).convert("RGB")
+    # check if file has been uploaded
+    if uploaded_file is not None:
+        # load the image
+        image = Image.open(uploaded_file).convert("RGB")
 
-    # context as prompt
-    prompt = generate_captions(img)
-    st.write("The Context is:", prompt)
+        # context as prompt
+        prompt = generate_captions(uploaded_file)
+        st.write("The Context is:", prompt)
 
-    # display the image
-    st.image(uploaded_file)
+        # display the image
+        st.image(image)
     
     # Generate button
     if st.button("Generate"):