Spaces:
Running
on
Zero
Running
on
Zero
Anton Bushuiev
commited on
Commit
·
bee03d0
1
Parent(s):
3a7f59d
Apply spaces.GPU to the whole predict logic
Browse files
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)
|