Update app.py

app.py

@@ -18,18 +18,21 @@ model.eval()
 st.title('Zero-shot Object Detection')
 
 # Input image and query image upload
-col1, col2 = st.beta_columns(2)
+col1, col2 = st.columns(2)
 with col1:
     uploaded_image = st.file_uploader("Upload input image", type=["jpg", "jpeg", "png"])
     if uploaded_image is not None:
         image = Image.open(uploaded_image)
         st.image(image, caption='Input Image', use_column_width=True)
-
+    else:
+        st.image('1.png', caption='Input Image', use_column_width=True)
 with col2:
     uploaded_query = st.file_uploader("Upload query image", type=["jpg", "jpeg", "png"])
     if uploaded_query is not None:
         query_image = Image.open(uploaded_query)
         st.image(query_image, caption='Query Image', use_column_width=True)
+    else:
+        st.image('0.png', caption='Input Image', use_column_width=True)
 
 # Threshold ratio bar and class name input
 threshold_ratio = st.slider('Select threshold ratio:', min_value=0.0, max_value=1.0, step=0.1, value=0.6)