app

app.py

@@ -43,7 +43,7 @@ topK = 5
 T2I = "以文搜图"
 I2I = "以图搜图"
 
-DDT = "双塔动态嵌入"
+DPDT = "双塔动态池化"
 UEFDT = "双塔联合融合"
 IEFDT = "双塔嵌入融合"
 ViLT = "视觉语言预训练"
@@ -99,7 +99,7 @@ def search(mode, method, image, text):
     if method == ViLT: 
         pass
     else:
-        if method == DDT: swap_width = 5
+        if method == DPDT: swap_width = 5
         elif method == UEFDT: swap_width = 2
         elif method == IEFDT: swap_width = 1
 
@@ -152,7 +152,7 @@ if __name__ == "__main__":
         fn=search,
         inputs=[
             gr.inputs.Radio([I2I, T2I]),
-            gr.inputs.Radio([DDT, UEFDT, IEFDT, ViLT]),
+            gr.inputs.Radio([DPDT, UEFDT, IEFDT, ViLT]),
             gr.inputs.Image(shape=(400, 400), label="Image to search", optional=True),
             gr.inputs.Textbox(
                 lines=1, label="Text query", placeholder="please input text query here...",
@@ -167,7 +167,7 @@ if __name__ == "__main__":
         gr.outputs.Image(type="auto", label="5rd Best match")
         ],
         examples=[
-            [I2I, DDT, cat_image, ""],
+            [I2I, DPDT, cat_image, ""],
             [I2I, ViLT, dog_image, ""],
             [T2I, UEFDT, w1_image, "a woman is walking on the road"],
             [T2I, IEFDT, w2_image, "a boy is eating apple"],