Spaces:

C2MV
/

Biotech2

Running on Zero

C2MV commited on Sep 22

Commit

ac2323a

•

1 Parent(s): 6e8da07

Update decorators.py

Files changed (1) hide show

decorators.py CHANGED Viewed

@@ -1,11 +1,11 @@
 # decorators.py
-import spaces
 # Decorador personalizado que usa GPU si está disponible
 def gpu_decorator(duration=100):
     def decorator(func):
-        @spaces.GPU(duration=duration)
         def wrapper(*args, **kwargs):
             return func(*args, **kwargs)
         return wrapper

 # decorators.py
+from spaces import GPU  # Asegúrate de que 'spaces' está correctamente importado desde Hugging Face
 # Decorador personalizado que usa GPU si está disponible
 def gpu_decorator(duration=100):
     def decorator(func):
+        @GPU(duration=duration)
         def wrapper(*args, **kwargs):
             return func(*args, **kwargs)
         return wrapper