Update run.py

run.py

@@ -35,7 +35,7 @@ def flip_periodically(im, interval_ms=0):
     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().numpy()
+    logits = outputs.logits.cpu().detach().numpy()
     sleep(interval_ms / 1000)  # Convert milliseconds to seconds
     return logits[0, 0]  #np.flipud(im)