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)
|