Update run.py

run.py

@@ -38,12 +38,11 @@ model.to(device).eval()
 # counter = Count()
 
 def segment(im, interval_s=2):
-    global counter
-    counter.increment()
+    
     im = cv2.resize(im, (im.shape[1] // 2, im.shape[0] // 2))
     imout = im.copy()
     
-    if cnt % 10 == 0:
+    if counter.increment() % 3 == 0:
         # if (counter.imout.sum() == 0) or ((cnt % 100) == 0):
         pixel_values = image_processor(im, return_tensors="pt").pixel_values.to(device)
         outputs = model(pixel_values=pixel_values)