add app.py

app.py

@@ -9,16 +9,6 @@ from tqdm import tqdm
 MODELS = ["wrice/waveunet-vctk-48khz", "wrice/waveunet-vctk-24khz"]
 
 
-def main():
-    """Main."""
-    iface = gr.Interface(
-        fn=denoise,
-        inputs=[gr.Dropdown(choices=MODELS, default=MODELS[0]), "audio"],
-        outputs="audio",
-    )
-    iface.launch()
-
-
 def denoise(model_name, inputs):
     """Denoise audio."""
     model = WaveUNetModel.from_pretrained(model_name)
@@ -52,5 +42,9 @@ def denoise(model_name, inputs):
     return model.config.sample_rate, denoised
 
 
-if __name__ == "__main__":
-    main()
+iface = gr.Interface(
+    fn=denoise,
+    inputs=[gr.Dropdown(choices=MODELS, default=MODELS[0]), "audio"],
+    outputs="audio",
+)
+iface.launch()