Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ from utils import replace_pixels_if_difference_exceeds, get_mask, remove_waterma
|
|
| 34 |
|
| 35 |
# --- MAIN LAMA PROCESSING ---
|
| 36 |
@spaces.GPU
|
| 37 |
-
def run_lama(image_path, n=
|
| 38 |
# Create temp directories
|
| 39 |
temp_dir = tempfile.mkdtemp()
|
| 40 |
data_dir = os.path.join(temp_dir, 'data_for_prediction')
|
|
@@ -136,7 +136,7 @@ if __name__ == "__main__":
|
|
| 136 |
n_slider = gr.Slider(
|
| 137 |
minimum=5,
|
| 138 |
maximum=100,
|
| 139 |
-
value=
|
| 140 |
step=1,
|
| 141 |
label="🎨 Pixel Difference Threshold (n)",
|
| 142 |
info="Higher = more original pixels preserved. Lower = more LAMA inpainting kept."
|
|
@@ -155,9 +155,9 @@ if __name__ == "__main__":
|
|
| 155 |
|
| 156 |
gr.Markdown("""
|
| 157 |
### 💡 Tips:
|
| 158 |
-
- **n =
|
| 159 |
-
- **n <
|
| 160 |
-
- **n >
|
| 161 |
""")
|
| 162 |
|
| 163 |
demo.launch()
|
|
|
|
| 34 |
|
| 35 |
# --- MAIN LAMA PROCESSING ---
|
| 36 |
@spaces.GPU
|
| 37 |
+
def run_lama(image_path, n=45): # ← ADD n PARAMETER
|
| 38 |
# Create temp directories
|
| 39 |
temp_dir = tempfile.mkdtemp()
|
| 40 |
data_dir = os.path.join(temp_dir, 'data_for_prediction')
|
|
|
|
| 136 |
n_slider = gr.Slider(
|
| 137 |
minimum=5,
|
| 138 |
maximum=100,
|
| 139 |
+
value=45,
|
| 140 |
step=1,
|
| 141 |
label="🎨 Pixel Difference Threshold (n)",
|
| 142 |
info="Higher = more original pixels preserved. Lower = more LAMA inpainting kept."
|
|
|
|
| 155 |
|
| 156 |
gr.Markdown("""
|
| 157 |
### 💡 Tips:
|
| 158 |
+
- **n = 45 (default)**: Balanced result
|
| 159 |
+
- **n < 50**: Removees watermark more (but may blur details)
|
| 160 |
+
- **n > 50**: Prefer original pixels (sharper, but watermark may peek through)
|
| 161 |
""")
|
| 162 |
|
| 163 |
demo.launch()
|