Update app.py

app.py

@@ -29,12 +29,16 @@ image = None  # Initialize image variable
 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)
+        try:
+            image = Image.open(file).resize((256, 256), Image.LANCZOS)
+        except Exception as e:
+            st.error(f"Error: {e}")
 
 elif opt == 'Upload image via link':
     try:
         img = st.text_input('Enter the Image Address')
-        image = Image.open(urllib.request.urlopen(img)).resize((256, 256), Image.LANCZOS)
+        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!")