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__":
|