Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
mariesig commited on
Commit ·
377e7f5
1
Parent(s): bc483dc
rmv gradio created audio
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ import time
|
|
| 3 |
from typing import Optional, Any
|
| 4 |
|
| 5 |
import gradio as gr
|
|
|
|
| 6 |
from loguru import logger
|
| 7 |
from PIL import Image
|
| 8 |
|
|
@@ -144,6 +145,7 @@ def delete_enhancement(path_to_delete: str):
|
|
| 144 |
Consider adding a flag to files to indicate whether files can be deleted, if disk usage becomes an issue.
|
| 145 |
"""
|
| 146 |
if not path_to_delete or not os.path.exists(path_to_delete):
|
|
|
|
| 147 |
logger.warning(f"Path to delete is invalid or does not exist: {path_to_delete}")
|
| 148 |
return
|
| 149 |
filename_no_ext = os.path.splitext(os.path.basename(path_to_delete))[0]
|
|
@@ -188,7 +190,9 @@ def start_processing(sample_path: str, model_radio: str, api_key: str) -> tuple[
|
|
| 188 |
return gr.update(visible=False), gr.update(interactive=False), input_enhancement_path
|
| 189 |
|
| 190 |
|
| 191 |
-
def enable_new_input():
|
|
|
|
|
|
|
| 192 |
return gr.update(visible=True), gr.update(interactive=True)
|
| 193 |
|
| 194 |
|
|
@@ -290,7 +294,7 @@ with gr.Blocks() as demo:
|
|
| 290 |
api_key,
|
| 291 |
],
|
| 292 |
outputs=[noisy_audio, noisy_image, enhanced_audio, enhanced_image],
|
| 293 |
-
).then(enable_new_input,
|
| 294 |
noise_type.change(toggle_SNR, noise_type, noise_level)
|
| 295 |
|
| 296 |
cleanup_tmp(minutes_keep=0, filter=[])
|
|
|
|
| 3 |
from typing import Optional, Any
|
| 4 |
|
| 5 |
import gradio as gr
|
| 6 |
+
from httpx import delete
|
| 7 |
from loguru import logger
|
| 8 |
from PIL import Image
|
| 9 |
|
|
|
|
| 145 |
Consider adding a flag to files to indicate whether files can be deleted, if disk usage becomes an issue.
|
| 146 |
"""
|
| 147 |
if not path_to_delete or not os.path.exists(path_to_delete):
|
| 148 |
+
cleanup_tmp(minutes_keep=120, filter=EXAMPLES)
|
| 149 |
logger.warning(f"Path to delete is invalid or does not exist: {path_to_delete}")
|
| 150 |
return
|
| 151 |
filename_no_ext = os.path.splitext(os.path.basename(path_to_delete))[0]
|
|
|
|
| 190 |
return gr.update(visible=False), gr.update(interactive=False), input_enhancement_path
|
| 191 |
|
| 192 |
|
| 193 |
+
def enable_new_input(old_input: str):
|
| 194 |
+
os.remove(old_input)
|
| 195 |
+
print(f"Removed old input file: {old_input}")
|
| 196 |
return gr.update(visible=True), gr.update(interactive=True)
|
| 197 |
|
| 198 |
|
|
|
|
| 294 |
api_key,
|
| 295 |
],
|
| 296 |
outputs=[noisy_audio, noisy_image, enhanced_audio, enhanced_image],
|
| 297 |
+
).then(enable_new_input, audio_file, [examples_group, btn])
|
| 298 |
noise_type.change(toggle_SNR, noise_type, noise_level)
|
| 299 |
|
| 300 |
cleanup_tmp(minutes_keep=0, filter=[])
|