Anton Bushuiev commited on
Commit
bee03d0
·
1 Parent(s): 3a7f59d

Apply spaces.GPU to the whole predict logic

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -141,12 +141,11 @@ def setup():
141
  print("Setup complete")
142
 
143
 
144
- @spaces.GPU
145
  def _predict_gpu(msdata):
146
  embs = dreams_embeddings(msdata)
147
  return embs
148
 
149
-
150
  def _predict_core(lib_pth, in_pth, progress):
151
  """Core prediction function without error handling"""
152
  in_pth = Path(in_pth)
 
141
  print("Setup complete")
142
 
143
 
 
144
  def _predict_gpu(msdata):
145
  embs = dreams_embeddings(msdata)
146
  return embs
147
 
148
+ @spaces.GPU
149
  def _predict_core(lib_pth, in_pth, progress):
150
  """Core prediction function without error handling"""
151
  in_pth = Path(in_pth)