matheus-erthal commited on
Commit
44e2df3
1 Parent(s): 83e292f

Ajustando recebimento das imagens

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ def greet(name):
5
  return "Hello " + name + "!!"
6
 
7
  def image_similarity(image1, image2):
8
- return ImageSimilarity(1).check([image1, image2])
9
 
10
- iface = gr.Interface(fn=image_similarity, inputs=["image", "image"], outputs="text")
11
  iface.launch()
 
5
  return "Hello " + name + "!!"
6
 
7
  def image_similarity(image1, image2):
8
+ return ImageSimilarity(1).check([image1.filename, image2.filename])
9
 
10
+ iface = gr.Interface(fn=image_similarity, inputs=[gr.inputs.Image(type="pil"), gr.inputs.Image(type="pil")], outputs="text")
11
  iface.launch()