Upload 2 files

app.py

@@ -29,7 +29,7 @@ def work11(image1, image2):
 demo = gr.Interface(title="图片相似度推理",
  css="",
  fn=work11,
- inputs=[gr.Image(type="filepath", label="图片1"), gr.Image(type="filepath", label="图片2")],
+ inputs=[gr.Image(type="pil", label="图片1"), gr.Image(type="pil", label="图片2")],
  outputs=[gr.Textbox(lines=3, label="推理结果")])
 #
 # # demo = gr.Interface(fn=work, inputs="image,text", outputs="text")

image_feature.py

@@ -141,10 +141,10 @@ def infer1(image1, image2):
  print("这是finally块")
 
 # 输出图片向量
-def xiang(url):
+def xiang(upload_image):
  try:
 
- image = Image.open(url).convert('RGB')
+ image = upload_image.convert('RGB')
 
  processor = AutoImageProcessor.from_pretrained('facebook/dinov2-giant')
  model = AutoModel.from_pretrained('facebook/dinov2-giant')