Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 import PIL
 import urllib.request
 from utils import *
-from fastai.basics import *
+from fastai.data.external import *
 
 # Function to classify the garbage
 def classify_garbage(img_path, model):
@@ -29,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 = to_image(file).resize((256, 256), Image.LANCZOS)
+        image = Image.open(file).resize((256, 256), Image.LANCZOS)
 
 elif opt == 'Upload image via link':
     img_url = st.text_input('Enter the Image Address')
     try:
-        image = to_image(urllib.request.urlopen(img_url)).resize((256, 256), Image.LANCZOS)
+        image = Image.open(urllib.request.urlopen(img_url)).resize((256, 256), Image.LANCZOS)
     except ValueError:
         st.error("Please Enter a valid Image Address!")