liruiw commited on
Commit
178cd5c
·
1 Parent(s): 92a983a

reactivate

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,9 +93,9 @@ def init_model():
93
 
94
  if __name__ == '__main__':
95
  genie = init_model()
96
-
97
  with gr.Blocks() as demo:
98
  genie_instance = gr.State({'genie': genie})
 
99
 
100
  with gr.Row():
101
  image_selector = gr.Dropdown(
 
93
 
94
  if __name__ == '__main__':
95
  genie = init_model()
 
96
  with gr.Blocks() as demo:
97
  genie_instance = gr.State({'genie': genie})
98
+ genie.device = "cuda"
99
 
100
  with gr.Row():
101
  image_selector = gr.Dropdown(