Update app.py

app.py

@@ -29,13 +29,14 @@ 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 = Image.open(file).resize((256, 256), Image.LANCZOS)
-
+        pil_image = Image.open(file)
+        image = preprocess(pil_image)
 
 elif opt == 'Upload image via link':
     img_url = st.text_input('Enter the Image Address')
     try:
-        image = Image.open(urllib.request.urlopen(img_url)).resize((256, 256), Image.LANCZOS)
+        pil_image = Image.open(urllib.request.urlopen(img_url))
+        image = preprocess(pil_image)
     except ValueError:
         st.error("Please Enter a valid Image Address!")
 
@@ -48,7 +49,7 @@ if 'image' in locals():  # Check if image variable exists
             st.info('Worked1')
             
             # Ensure image shape is correct and add batch dimension
-            img_array = preprocess(file)  # This should return an array of shape (1, 256, 256, 3)
+            img_array = preprocess(image)  # This should return an array of shape (1, 256, 256, 3)
 
             st.info('Worked2')