Hugging Face's logo

Spaces:
Mountchicken
/
Rex-Omni
Running on Zero

App Files Community
1
Mountchicken commited on
Commit
27ca051
·
verified ·
1 Parent(s): 9a6cc54

Update rex_omni/wrapper.py

Browse files
Files changed (1) hide show
  1. rex_omni/wrapper.py +2 -2
rex_omni/wrapper.py CHANGED
@@ -4,7 +4,7 @@
4
  """
5
  Main wrapper class for Rex Omni
6
  """
7
-
8
  import base64
9
  import json
10
  import math
@@ -191,7 +191,7 @@ class RexOmniWrapper:
191
  """
192
  High-level wrapper for Rex-Omni
193
  """
194
-
195
  def __init__(
196
  self,
197
  model_path: str,
 
4
  """
5
  Main wrapper class for Rex Omni
6
  """
7
+ import spaces
8
  import base64
9
  import json
10
  import math
 
191
  """
192
  High-level wrapper for Rex-Omni
193
  """
194
+ @spaces.GPU
195
  def __init__(
196
  self,
197
  model_path: str,