Update app.py

app.py

@@ -122,10 +122,15 @@ def format_bpe_display(bpe):
     # 使用HTML标签来设置字体大小、颜色，加粗，并居中
     return f"<div style='text-align:center; font-size:20px;'><strong>Current BPE: <span style='color:red;'>{bpe}</span></strong></div>"
 
-@spaces.GPU
+# def update_slider_index(x):
+#     global current_vis, current_bpe, current_index
+#     print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(current_bpe)}")
+#     if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe):
+#         return current_vis[x], format_bpe_display(current_bpe[x])
+#     else:
+#         return None, "索引超出范围"
+
 def update_slider_index(x):
-    global current_vis, current_bpe, current_index
-    print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(current_bpe)}")
     if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe):
         return current_vis[x], format_bpe_display(current_bpe[x])
     else:
@@ -172,18 +177,29 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
             bpe_display = gr.Markdown("Current BPE: ", visible=False)
 
     # 事件处理
+    # @spaces.GPU
+    # def on_run_clicked(model_type, image, text):
+    #     global current_vis, current_bpe, current_index
+    #     current_index = 0  # Reset index when new image is processed
+    #     image, vis, bpe = process_image(*load_model(model_type), model_type, image, text)
+    #     # Update the slider range and set value to 0
+    #     slider_max_val = len(current_bpe) - 1
+    #     bpe_text = format_bpe_display(bpe)
+    #     print("current_vis",len(current_vis))
+    #     print("current_bpe",len(current_bpe))
+    #     return image, vis, bpe_text, slider_max_val
     @spaces.GPU
     def on_run_clicked(model_type, image, text):
         global current_vis, current_bpe, current_index
         current_index = 0  # Reset index when new image is processed
-        image, vis, bpe = process_image(*load_model(model_type), model_type, image, text)
-        # Update the slider range and set value to 0
+        model, tokenizer, transform, device = load_model(model_type)
+        image, vis, bpe = process_image(model, tokenizer, transform, device, model_type, image, text)
+        current_vis = vis
+        current_bpe = bpe
         slider_max_val = len(current_bpe) - 1
-        bpe_text = format_bpe_display(bpe)
-        print("current_vis",len(current_vis))
-        print("current_bpe",len(current_bpe))
-        return image, vis, bpe_text, slider_max_val
-
+        bpe_text = format_bpe_display(current_bpe[current_index])
+        return image, current_vis[current_index], bpe_text, slider_max_val
+    
     run_btn.click(
         on_run_clicked,
         inputs=[model_type, image_input, text_input],
@@ -210,11 +226,16 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     outputs=[heatmap, bpe_display]
     # )
 
+    # index_slider.change(
+    #         update_slider_index,
+    #         inputs=index_slider,
+    #         outputs=[heatmap, bpe_display]
+    #     )
     index_slider.change(
-            update_slider_index,
-            inputs=index_slider,
-            outputs=[heatmap, bpe_display]
-        )
+    update_slider_index,
+    inputs=[index_slider],
+    outputs=[heatmap, bpe_display])
+
 
 if __name__ == "__main__":
     demo.launch()