add sleep=2sec

run.py

@@ -20,7 +20,7 @@ model.load_state_dict(torch.load(weights2load, weights_only=True, map_location=d
 model.to(device).eval()
 
 
-def flip_periodically(im, interval_ms=0):
+def flip_periodically(im, interval_s=2):
     """
     Flips the image periodically with the given interval.
 
@@ -36,7 +36,7 @@ def flip_periodically(im, interval_ms=0):
     pixel_values = image_processor(im, return_tensors="pt").pixel_values.to(device)
     outputs = model(pixel_values=pixel_values)
     logits = outputs.logits.cpu().detach().numpy() ** 2
-    sleep(interval_ms / 1000)  # Convert milliseconds to seconds
+    sleep(interval_s)
     imout = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
     return imout  #np.flipud(im)