Upload 2 files

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import image_feature as func
 
 
-def work(image1, image2):
+def work11(image1, image2):
     return func.infer1(image1, image2)
 
 
@@ -15,7 +15,7 @@ def work(image1, image2):
 # demo.launch()
 
 # 定义你的界面
-with gr.Interface(fn=work,
+with gr.Interface(fn=work11,
                   inputs=[gr.Textbox(label='图片1', lines=1), gr.Textbox(label='图片2', lines=1)],  # 两个文本输入框
                   outputs=[gr.Textbox(lines=3, label="推理结果")],  # 输出为文本
                   title="图片相似度推理",  # 界面标题

image_feature.py

@@ -54,9 +54,9 @@ model = AutoModel.from_pretrained("google/vit-base-patch16-224").to(DEVICE)
 
 
 #  推理
-def infer(img):
+def infer2(url):
     # image_real = Image.open(requests.get(img_urls[0], stream=True).raw).convert("RGB")
-    image = Image.open(requests.get(img, stream=True).raw).convert("RGB")
+    image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
     inputs = processor(image, return_tensors="pt").to(DEVICE)
     outputs = model(**inputs)
     return outputs.pooler_output
@@ -64,8 +64,8 @@ def infer(img):
 
 #  推理
 def infer1(image1, image2):
-    embed_real = infer(image1)
-    embed_gen = infer(image2)
+    embed_real = infer2(image1)
+    embed_gen = infer2(image2)
     similarity_score = cosine_similarity(embed_real, embed_gen, dim=1)
     print(similarity_score)
     # 如果你想在CPU上操作这个值，你需要先将tensor移动到CPU