Spaces:

minh9972t12
/

yolocar

Sleeping

minh9972t12 commited on Sep 3

Commit

7cc748c

1 Parent(s): 403fdd5

Update src/comparison.py

Files changed (1) hide show

src/comparison.py CHANGED Viewed

@@ -145,7 +145,7 @@ class DamageComparator:
             roi_rgb = cv2.cvtColor(roi, cv2.COLOR_BGR2RGB)
             # Process with CLIP
-            inputs = self.clip_processor(images=roi_rgb, return_tensors="pt")
             inputs = {k: v.to(self.device) for k, v in inputs.items()}
             with torch.no_grad():

             roi_rgb = cv2.cvtColor(roi, cv2.COLOR_BGR2RGB)
             # Process with CLIP
+            inputs = self.clip_processor(images=roi_rgb, return_tensors="pt", use_fast=True)
             inputs = {k: v.to(self.device) for k, v in inputs.items()}
             with torch.no_grad():