Update app.py

app.py

@@ -64,6 +64,18 @@ if uploaded_file is not None:
         bytes_data = uploaded_file.getvalue()
         img_input = Image.open(BytesIO(bytes_data)).convert("RGBA")
 
+    # Resize the image while maintaining aspect ratio
+    max_size = 2000
+    img_width, img_height = img_input.size
+    if img_width > max_size or img_height > max_size:
+        if img_width > img_height:
+            new_width = max_size
+            new_height = int((max_size / img_width) * img_height)
+        else:
+            new_height = max_size
+            new_width = int((max_size / img_height) * img_width)
+        img_input = img_input.resize((new_width, new_height))
+    
     stroke_width = st.slider("Brush size", 1, 100, 50)
 
     st.write("**Now draw (brush) the part of image that you want to remove.**")
@@ -121,6 +133,6 @@ if uploaded_file is not None:
                 fmt="jpg",
                 label="Download Image"
             )
-                
+            
             st.info("**TIP**: If the result is not perfect, you can download it then "
                     "upload then remove the artifacts.")