Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ prompt = "What action should the robot take to pick the cup?"
|
|
52 |
inputs = processor(images=[image], text=prompt, return_tensors="pt")
|
53 |
generation_outputs = model.predict_action(inputs)
|
54 |
|
55 |
-
actions = processor.decode_actions(generation_outputs, unnorm_key="
|
56 |
print(actions)
|
57 |
```
|
58 |
|
|
|
52 |
inputs = processor(images=[image], text=prompt, return_tensors="pt")
|
53 |
generation_outputs = model.predict_action(inputs)
|
54 |
|
55 |
+
actions = processor.decode_actions(generation_outputs, unnorm_key="fractal20220817_data/0.1.0")
|
56 |
print(actions)
|
57 |
```
|
58 |
|