aryrk commited on
Commit
30787ce
·
1 Parent(s): c4716b6

[fix error]

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -64,13 +64,15 @@ sample_images = [
64
 
65
  iface = gr.Interface(
66
  fn=reflection_removal,
67
- inputs=gr.File(label="Upload Image", type="filepath"),
 
 
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__":