Upload folder using huggingface_hub
Browse files- test_huggingface.py +1 -1
test_huggingface.py
CHANGED
|
@@ -19,7 +19,7 @@ if __name__ == "__main__":
|
|
| 19 |
|
| 20 |
image = Image.open("example.png").convert("RGB")
|
| 21 |
images = [image] * args.num_images
|
| 22 |
-
prompt = "What action should the robot take to pick the
|
| 23 |
inputs = processor(images=images, text=prompt, unnorm_key="bridge_orig/1.0.0", return_tensors="pt")
|
| 24 |
print(inputs)
|
| 25 |
|
|
|
|
| 19 |
|
| 20 |
image = Image.open("example.png").convert("RGB")
|
| 21 |
images = [image] * args.num_images
|
| 22 |
+
prompt = "What action should the robot take to pick the cup?"
|
| 23 |
inputs = processor(images=images, text=prompt, unnorm_key="bridge_orig/1.0.0", return_tensors="pt")
|
| 24 |
print(inputs)
|
| 25 |
|