Update app.py

app.py

@@ -3,7 +3,8 @@ import streamlit as st
 import numpy as np
 import PIL
 import urllib.request
-from utils import *  # Assuming the gen_labels() and preprocess() functions are in this module
+from utils import *
+from fastai.basics import *
 
 # Function to classify the garbage
 def classify_garbage(img_path, model):
@@ -28,12 +29,12 @@ opt = st.selectbox("How do you want to upload the image for classification?", ('
 if opt == 'Upload image from device':
     file = st.file_uploader('Select', type=['jpg', 'png', 'jpeg'])
     if file is not None:
-        image = PIL.Image.open(file).resize((256, 256), Image.LANCZOS)
+        image = to_image(file).resize((256, 256), Image.LANCZOS)
 
 elif opt == 'Upload image via link':
     img_url = st.text_input('Enter the Image Address')
     try:
-        image = PIL.Image.open(urllib.request.urlopen(img_url)).resize((256, 256), Image.LANCZOS)
+        image = to_image(urllib.request.urlopen(img_url)).resize((256, 256), Image.LANCZOS)
     except ValueError:
         st.error("Please Enter a valid Image Address!")