Update app.py

app.py

@@ -1,3 +1,9 @@
+import os
+import numpy as np
+from PIL import Image, ImageDraw
+
+
+import torch
 import torch.nn.functional as F
 import torchvision.transforms.functional as TF
 from transformers import AutoModel # trust_remote_code=True
@@ -211,6 +217,8 @@ def click_two_image_similarity(state1: dict, state2: dict, click_xy: tuple[int,
 # ----------------------------
 # Gradio UI
 # ----------------------------
+
+
 with gr.Blocks(theme=gr.themes.Soft(), title="DINOv3 Two‑Image Patch Similarity") as demo:
     gr.Markdown("# DINOv3 Two‑Image Patch Similarity")
     gr.Markdown("Upload two images, process, then click on image 1 to see similarities on both.")
@@ -234,4 +242,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="DINOv3 Two‑Image Patch Similarit
     # (rest of app: outputs, event wiring, functions, unchanged)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()