Hugging Face's logo

Spaces:
liruiw
/
hma
Runtime error

App Files Community
1
liruiw commited on
Commit
2c0d4d5
·
1 Parent(s): bd39096

updatE

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ genie = GenieSimulator(
74
  prompt_horizon=PROMPT_HORIZON,
75
  action_stride=1,
76
  domain='language_table',
77
- device="cpu"
78
  )
79
 
80
  image = Image.open("sim/assets/langtable_prompt/frame_06.png")
@@ -86,7 +86,7 @@ prompt_action = np.zeros(
86
  ).astype(np.float32)
87
  genie.set_initial_state((prompt_image, prompt_action))
88
  genie.device = "cuda"
89
- image = genie.reset()
90
 
91
 
92
  if __name__ == '__main__':
 
74
  prompt_horizon=PROMPT_HORIZON,
75
  action_stride=1,
76
  domain='language_table',
77
+ device="cuda"
78
  )
79
 
80
  image = Image.open("sim/assets/langtable_prompt/frame_06.png")
 
86
  ).astype(np.float32)
87
  genie.set_initial_state((prompt_image, prompt_action))
88
  genie.device = "cuda"
89
+
90
 
91
 
92
  if __name__ == '__main__':