Update app.py

app.py

@@ -3,48 +3,47 @@ import streamlit as st
 import numpy as np
 from PIL import Image
 import urllib.request
-from utils import *
 import io
+from utils import *
 
+# Initialize labels and model
 labels = gen_labels()
+model = model_arc()  # Assuming this function initializes and returns a trained model
 
-html_temp = '''
+# Streamlit UI
+st.markdown('''
     <div style="padding-bottom: 20px; padding-top: 20px; padding-left: 5px; padding-right: 5px">
     <center><h1>Garbage Segregation</h1></center>
     </div>
-    '''
-st.markdown(html_temp, unsafe_allow_html=True)
+''', unsafe_allow_html=True)
 
-html_temp = '''
+st.markdown('''
     <div>
-    <h2></h2>
     <center><h3>Please upload Waste Image to find its Category</h3></center>
     </div>
-    '''
-st.markdown(html_temp, unsafe_allow_html=True)
+''', unsafe_allow_html=True)
 
-opt = st.selectbox("How do you want to upload the image for classification?\n", ('Please Select', 'Upload image via link', 'Upload image from device'))
-
-image = None  # Initialize image variable
+opt = st.selectbox("How do you want to upload the image for classification?", 
+                   ('Please Select', 'Upload image via link', 'Upload image from device'))
 
+# Image processing based on user selection
+image = None
 if opt == 'Upload image from device':
     file = st.file_uploader('Select', type=['jpg', 'png', 'jpeg'])
-    if file is not None:
+    if file:
         try:
             image = Image.open(io.BytesIO(file.read())).resize((256, 256), Image.LANCZOS)
         except Exception as e:
-            st.error(f"Error: {e}")
+            st.error(f"Error reading the file: {e}")
 
 elif opt == 'Upload image via link':
-    try:
-        img = st.text_input('Enter the Image Address')
-        response = urllib.request.urlopen(img)
-        image = Image.open(response).resize((256, 256), Image.LANCZOS)
-    except ValueError:
-        if st.button('Submit'):
-            show = st.error("Please Enter a valid Image Address!")
-            time.sleep(4)
-            show.empty()
+    img_url = st.text_input('Enter the Image Address')
+    if st.button('Submit'):
+        try:
+            response = urllib.request.urlopen(img_url)
+            image = Image.open(response).resize((256, 256), Image.LANCZOS)
+        except ValueError:
+            st.error("Please Enter a valid Image Address!")
 
 try:
   if image is not None: