push app

app.py

@@ -1,7 +1,7 @@
 import os
 import numpy as np
 import streamlit as st
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFilter
 import numpy as np
 import torch
 
@@ -23,7 +23,7 @@ FASTSAM_CHECKPOINT = "FastSAM-x.pt"  # file name of the FastSAM model weights
 SD_MODEL_ID = "runwayml/stable-diffusion-inpainting"  # HF Hub model for SD Inpainting v1.5
 
 # Helper function: center crop and resize to 768x512 (landscape)
-def crop_resize_image(image, target_width=512, target_height=768):
+def crop_resize_image(image, target_width=480, target_height=640):
     desired_ratio = target_width / target_height  # 768/512 = 1.5
     width, height = image.size
     current_ratio = width / height
@@ -100,7 +100,7 @@ if "img" not in st.session_state:
     picture = st.camera_input("Take a picture", disabled=not enable)
     if picture is not None:
         img = Image.open(picture)
-        img = crop_resize_image(img, target_width=512, target_height=768)
+        img = crop_resize_image(img, target_width=480, target_height=640)
         st.session_state.img = img
         # Reset coordinates list on new capture
         st.session_state.coords_list = []