Update app.py

app.py

@@ -3,20 +3,24 @@ import torch
 import numpy as np
 from transformers import OwlViTProcessor, OwlViTForObjectDetection
 from torchvision import transforms
-from PIL import Image
+from PIL import Image, ImageDraw
 import cv2
 import torch.nn.functional as F
 import tempfile
 import os
 from SuperGluePretrainedNetwork.models.matching import Matching
 from SuperGluePretrainedNetwork.models.utils import read_image
+import matplotlib.pyplot as plt
+import matplotlib.cm as cm
+
+# Set device
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load models
 mixin = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch32")
 processor = OwlViTProcessor.from_pretrained("google/owlvit-base-patch32")
 model = mixin.to(device)
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 matching = Matching({
     'superpoint': {'nms_radius': 4, 'keypoint_threshold': 0.005, 'max_keypoints': 1024},
     'superglue': {'weights': 'outdoor', 'sinkhorn_iterations': 20, 'match_threshold': 0.2}