Spaces:
Sleeping
Sleeping
aryrk
commited on
Commit
·
30787ce
1
Parent(s):
c4716b6
[fix error]
Browse files
app.py
CHANGED
|
@@ -64,13 +64,15 @@ sample_images = [
|
|
| 64 |
|
| 65 |
iface = gr.Interface(
|
| 66 |
fn=reflection_removal,
|
| 67 |
-
inputs=
|
|
|
|
|
|
|
| 68 |
outputs=gr.Gallery(label="Results after Reflection Removal"),
|
| 69 |
-
title="Reflection Remover with Pix2Pix",
|
| 70 |
-
description="Upload images to remove reflections using a Pix2Pix model.",
|
| 71 |
examples=[
|
| 72 |
os.path.join("sample_images", img) for img in os.listdir("sample_images") if img.endswith((".jpg", ".jpeg", ".png"))
|
| 73 |
],
|
|
|
|
|
|
|
| 74 |
)
|
| 75 |
|
| 76 |
if __name__ == "__main__":
|
|
|
|
| 64 |
|
| 65 |
iface = gr.Interface(
|
| 66 |
fn=reflection_removal,
|
| 67 |
+
inputs=[
|
| 68 |
+
gr.Image(type="filepath", label="Upload Image (JPG/PNG)", live=True)
|
| 69 |
+
],
|
| 70 |
outputs=gr.Gallery(label="Results after Reflection Removal"),
|
|
|
|
|
|
|
| 71 |
examples=[
|
| 72 |
os.path.join("sample_images", img) for img in os.listdir("sample_images") if img.endswith((".jpg", ".jpeg", ".png"))
|
| 73 |
],
|
| 74 |
+
title="Reflection Remover with Pix2Pix",
|
| 75 |
+
description="Upload images to remove reflections using a Pix2Pix model. You can also try the sample images below."
|
| 76 |
)
|
| 77 |
|
| 78 |
if __name__ == "__main__":
|