Spaces:
Running
on
Zero
Running
on
Zero
roychao19477
commited on
Commit
·
776c247
1
Parent(s):
b65382e
Update figs
Browse files
app.py
CHANGED
|
@@ -48,7 +48,7 @@ model = SEMamba(cfg).to(device)
|
|
| 48 |
#model.eval()
|
| 49 |
|
| 50 |
@spaces.GPU
|
| 51 |
-
def enhance(filepath):
|
| 52 |
# Load model based on selection
|
| 53 |
ckpt_path = {
|
| 54 |
"VCTK-Demand": "ckpts/SEMamba_advanced.pth",
|
|
|
|
| 48 |
#model.eval()
|
| 49 |
|
| 50 |
@spaces.GPU
|
| 51 |
+
def enhance(filepath, model_name):
|
| 52 |
# Load model based on selection
|
| 53 |
ckpt_path = {
|
| 54 |
"VCTK-Demand": "ckpts/SEMamba_advanced.pth",
|