Spaces:

noamholz
/

gra

Sleeping

noamholz commited on Jan 20

Commit

e9e0a79

verified ·

1 Parent(s): d19df79

bug fix

Files changed (1) hide show

run.py CHANGED Viewed

@@ -31,7 +31,7 @@ def flip_periodically(im, interval_ms=0):
     Returns:
         The flipped image.
     """
-    pixel_values = image_processor(image, return_tensors="pt").pixel_values.to(device)
     outputs = model(pixel_values=pixel_values)
     logits = outputs.logits.cpu()
     sleep(interval_ms / 1000)  # Convert milliseconds to seconds

     Returns:
         The flipped image.
     """
+    pixel_values = image_processor(im, return_tensors="pt").pixel_values.to(device)
     outputs = model(pixel_values=pixel_values)
     logits = outputs.logits.cpu()
     sleep(interval_ms / 1000)  # Convert milliseconds to seconds