Spaces:
Running
on
Zero
Running
on
Zero
roychao19477
commited on
Commit
·
6ecd7dd
1
Parent(s):
d402093
Upload
Browse files
app.py
CHANGED
|
@@ -53,8 +53,6 @@ sdict = torch.load(ckpt, map_location=device)
|
|
| 53 |
model.load_state_dict(sdict["generator"])
|
| 54 |
model.eval()
|
| 55 |
|
| 56 |
-
|
| 57 |
-
|
| 58 |
|
| 59 |
@spaces.GPU
|
| 60 |
def enhance(filepath):
|
|
@@ -91,6 +89,7 @@ def enhance(filepath):
|
|
| 91 |
|
| 92 |
|
| 93 |
with gr.Blocks(theme="soft") as demo:
|
|
|
|
| 94 |
se_demo = gr.Interface(
|
| 95 |
fn = enhance,
|
| 96 |
inputs = [
|
|
@@ -102,7 +101,7 @@ with gr.Blocks(theme="soft") as demo:
|
|
| 102 |
#gr.Plot(label="Spectrogram")
|
| 103 |
],
|
| 104 |
title = "SEMamba",
|
| 105 |
-
cache_examples=False
|
| 106 |
)
|
| 107 |
|
| 108 |
if __name__ == "__main__":
|
|
|
|
| 53 |
model.load_state_dict(sdict["generator"])
|
| 54 |
model.eval()
|
| 55 |
|
|
|
|
|
|
|
| 56 |
|
| 57 |
@spaces.GPU
|
| 58 |
def enhance(filepath):
|
|
|
|
| 89 |
|
| 90 |
|
| 91 |
with gr.Blocks(theme="soft") as demo:
|
| 92 |
+
gr.HTML("<h1><center>SEMamba</center></h1>")
|
| 93 |
se_demo = gr.Interface(
|
| 94 |
fn = enhance,
|
| 95 |
inputs = [
|
|
|
|
| 101 |
#gr.Plot(label="Spectrogram")
|
| 102 |
],
|
| 103 |
title = "SEMamba",
|
| 104 |
+
cache_examples=False,
|
| 105 |
)
|
| 106 |
|
| 107 |
if __name__ == "__main__":
|