Upload image_feature.py

image_feature.py

@@ -64,14 +64,21 @@ def infer2(url):
 
 #  推理
 def infer1(image1, 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
-    t_cpu = similarity_score.cpu()
+    try:
+        embed_real = infer2(image1)
+        embed_gen = infer2(image2)
+        similarity_score = cosine_similarity(embed_real, embed_gen, dim=1)
+        print(similarity_score)
+        # 如果你想在CPU上操作这个值，你需要先将tensor移动到CPU
+        t_cpu = similarity_score.cpu()
 
-    # 然后提取这个值
-    return t_cpu.item()
+        # 然后提取这个值
+        return t_cpu.item()
+
+    except Exception as e:
+        print(f"发生了一个错误: {e}")
+    finally:
+        # 无论是否发生异常，都会执行此代码块
+        print("这是finally块")
 
 # tensor([0.6061], device='cuda:0', grad_fn=<SumBackward1>)