Spaces:

noamholz
/

gra

Sleeping

noamholz commited on Jan 20

Commit

f9b6fc2

verified ·

1 Parent(s): be1f66a

Update run.py

Files changed (1) hide show

run.py CHANGED Viewed

@@ -33,7 +33,7 @@ def flip_periodically(im, interval_ms=0):
     """
     transforms.ToTensor()(im)
-    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

     """
     transforms.ToTensor()(im)
+    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