fix a bug in metauas

app.py

@@ -91,9 +91,12 @@ def process_image(prompt_img, query_img, options):
     query_img = test_data["query_image"][0] * 255
     query_img = query_img.permute(1,2,0)
 
-    anomaly_map = predicted_masks.squeeze().detach()[:, :, None].cpu().numpy().repeat(3, 2)
-    
-    anomaly_map_vis = apply_ad_scoremap(query_img.cpu(), normalize(anomaly_map))
+    if device == 'cuda':
+        anomaly_map = predicted_masks.squeeze().detach()[:, :, None].cpu().numpy().repeat(3, 2)
+        anomaly_map_vis = apply_ad_scoremap(query_img.cpu(), normalize(anomaly_map))
+    else:
+        anomaly_map = predicted_masks.squeeze().detach()[:, :, None].numpy().repeat(3, 2)
+        anomaly_map_vis = apply_ad_scoremap(query_img, normalize(anomaly_map))
 
 
     anomaly_map = (anomaly_map * 255).astype(np.uint8)
@@ -151,7 +154,7 @@ demo.css = """
 }
 
 #submit-button:hover {
-    background-color: darkorange !important;  /* Darker orange on hover */
+    background-color: orange !important;  /* Darker orange on hover */
 }
 """