Aditya Sahu commited on
Commit
adaf99d
·
verified ·
1 Parent(s): 9477618

Add @spaces.GPU for ZeroGPU

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -27,6 +27,8 @@ def _resolve_uploaded_path(uploaded):
27
  return getattr(uploaded, attr)
28
  return None
29
 
 
 
30
  def compile_model(model_name, vmem_value, lpmem_value, uploaded_model):
31
  # Decide the source model path (uploaded has priority)
32
  uploaded_path = _resolve_uploaded_path(uploaded_model)
 
27
  return getattr(uploaded, attr)
28
  return None
29
 
30
+
31
+ @spaces.GPU
32
  def compile_model(model_name, vmem_value, lpmem_value, uploaded_model):
33
  # Decide the source model path (uploaded has priority)
34
  uploaded_path = _resolve_uploaded_path(uploaded_model)