some_improve

app.py

@@ -1,12 +1,13 @@
 from audio_denoiser.AudioDenoiser import AudioDenoiser
 from timeit import default_timer as timer
+from datetime import datetime
 import gradio as gr
 from torchaudio import AudioMetaData
 import torch
 import tempfile
-import os
-
+import os,pytz
 
+tz = pytz.timezone('Asia/Singapore')
 theme='remilia/Ghostly'
 device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')
 
@@ -15,21 +16,31 @@ def denoiser(win, auto_scale):
     if win is None:
       gr.Warning('Audio does not exist. Please ensure that the audio has been successfully uploaded.')
       return None,None
-
+    
     startTime=timer()
     denoiser = AudioDenoiser(device=device)
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.wav')
     wout = temp_file.name
     temp_file.close() 
-    denoiser.process_audio_file(win, wout, auto_scale)
+    try:
+      denoiser.process_audio_file(win, wout, auto_scale)
+    except RuntimeError as e:
+      gr.Warning(str(e))
+      return None,None
+    
     endTime=timer()
     info=(f'🆗Completion time: {round(endTime-startTime,4)}s')
+    now=datetime.now(tz).strftime('%H:%M:%S')
+    print(f'{now}-{info}')
     return wout,info
 
-
+examples = [
+    ["sample/exp1.wav", "sample/exp1_d.wav"],
+    ["sample/exp2.wav", "sample/exp2_d.wav"]
+]
 with gr.Blocks(theme=theme) as app:
     gr.HTML('''
-    <h1 style="font-size: 25px;">Audio Dnoiser</h1>
+    <h1 style="font-size: 25px;">Audio Denoiser</h1>
     <p style="margin-bottom: 10px; font-size: 100%">
  
 Originating from the project: <a href='https://github.com/jose-solorzano/audio-denoiser'>audio-denoiser</a><br>
@@ -47,8 +58,12 @@ model: <a href='https://huggingface.co/jose-h-solorzano/audio-denoiser-512-32-v1
     info=gr.Textbox(label='info')
 
     btn.click(
-  denoiser,
-  inputs=[audio_in, scale],
-  outputs=[audio_out,info])
+      denoiser,
+      inputs=[audio_in, scale],
+      outputs=[audio_out,info])
+
+    examples = gr.Examples(examples=examples, inputs=audio_in, outputs=audio_out, fn=denoiser)
+
+     
 
 app.launch(share=True)

sample/{noise2.wav → exp1.wav}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c8e4445aed300ec8954d00aed652fdc1043b6dcdbf9d896414125fa4b513a5b9
-size 441044
+oid sha256:4d79018dd06ac5bdcb27517d60ba30d61cf93b4842505d63b3986058224c6607
+size 1318990

sample/{noise1.wav → exp2.wav}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ad861d15fffc0a933a839cc4730b5e42d56a86084f686f130451c7134c8ed26f
-size 438100
+oid sha256:a8ca61a37ea9c3172629f71b36029eb2268b78542455f6e942bb7926e5fa2cf8
+size 737358

sample/noise3.wav

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b5664d57114d9d81d561919493b91fca1a994679fdbaec24ece60eb2fd406afa
-size 790316