Update app.py

app.py

@@ -16,7 +16,6 @@ import zipfile
 with zipfile.ZipFile("Vegetable Images.zip", 'r') as zip_ref:
     zip_ref.extractall('.')
 
-placeholder = st.empty()
 
 class ImageVectorizer:
     '''
@@ -119,27 +118,22 @@ def process_input_image(img_url):
 vectorizer = ImageVectorizer()
 collection = get_milvus_collection()
 
-
-def main():
-    try:
-        st.markdown("<h3>Find Similar Vegetable Images</h3>", unsafe_allow_html=True)
-        desc = '''<p style="font-size: 15px;">Allowed Vegetables: Broad Bean, Bitter Gourd, Bottle Gourd, 
-        Green Round Brinjal, Broccoli, Cabbage, Capsicum, Carrot, Cauliflower, Cucumber, 
-        Raw Papaya, Potato, Green Pumpkin, Radish, Tomato.
-        </p> 
-        <p style="font-size: 13px;">Image embeddings are extracted from a fine-tuned VGG model. The model is fine-tuned on <a href="https://www.kaggle.com/datasets/misrakahmed/vegetable-image-dataset" target="_blank">images</a> clicked using a mobile phone camera. 
-        Embeddings of 20,000 vegetable images are stored in Milvus vector database. Embeddings of the input image are computed and 15 most similar images (based on L2 distance) are displayed.</p>
-        '''
-        st.markdown(desc, unsafe_allow_html=True)
-        img_url = st.text_input("Paste the image URL of a vegetable and hit Enter:", "")
-        if img_url:
-            placeholder.empty()
-            img_path = process_input_image(img_url)
-            find_similar_images(img_path, 15)
-            delete_file(os.path.dirname(img_path))
-    except Exception as e:
-        st.error(f'An unexpected error occured:  \n{e}')
-
-
-if __name__ == "__main__":
-    main()
+try:
+    st.markdown("<h3>Find Similar Vegetable Images</h3>", unsafe_allow_html=True)
+    desc = '''<p style="font-size: 15px;">Allowed Vegetables: Broad Bean, Bitter Gourd, Bottle Gourd, 
+    Green Round Brinjal, Broccoli, Cabbage, Capsicum, Carrot, Cauliflower, Cucumber, 
+    Raw Papaya, Potato, Green Pumpkin, Radish, Tomato.
+    </p> 
+    <p style="font-size: 13px;">Image embeddings are extracted from a fine-tuned VGG model. The model is fine-tuned on <a href="https://www.kaggle.com/datasets/misrakahmed/vegetable-image-dataset" target="_blank">images</a> clicked using a mobile phone camera. 
+    Embeddings of 20,000 vegetable images are stored in Milvus vector database. Embeddings of the input image are computed and 15 most similar images (based on L2 distance) are displayed.</p>
+    '''
+    st.markdown(desc, unsafe_allow_html=True)
+    img_url = st.text_input("Paste the image URL of a vegetable and hit Enter:", "")
+    placeholder = st.empty()
+    if img_url:
+        placeholder.empty()
+        img_path = process_input_image(img_url)
+        find_similar_images(img_path, 15)
+        delete_file(os.path.dirname(img_path))
+except Exception as e:
+    st.error(f'An unexpected error occured:  \n{e}')