Upload gui.py with huggingface_hub

gui.py

@@ -468,12 +468,25 @@ def create_interface():
         download_ensemble.click(fn=download_audio, inputs=[url_ensemble], outputs=[ensemble_audio])
         ensemble_button.click(
             auto_ensemble_process,
-            inputs=[ensemble_audio, ensemble_models, ensemble_seg_size, ensemble_overlap, output_format, ensemble_use_tta, model_file_dir, output_dir, norm_threshold, amp_threshold, batch_size, ensemble_method, only_instrumental],
+            inputs=[ensemble_audio, ensemble_models, ensemble_seg_size, ensemble_overlap, out_format, ensemble_use_tta, model_file_dir, out_dir, norm_threshold, amp_threshold, batch_size, ensemble_method, only_instrumental],
             outputs=[ensemble_output, ensemble_status]
         )
     
     return app
 
 if __name__ == "__main__":
-    interface = create_interface()
-    interface.launch()
+    parser = argparse.ArgumentParser(description="Music Source Separation Web UI")
+    parser.add_argument("--port", type=int, default=7860, help="Port to run the UI on")
+    parser.add_argument("--ngrok-token", type=str, default=None, help="Ngrok token for tunneling")
+    args = parser.parse_args()
+
+    app = create_interface()  # app değişkenini burada tanımlıyoruz
+
+    if args.ngrok_token:
+        from pyngrok import ngrok
+        ngrok.set_auth_token(args.ngrok_token)
+        public_url = ngrok.connect(args.port, "http")
+        print(f"Ngrok URL: {public_url}")
+        app.launch(server_name="0.0.0.0", server_port=args.port)
+    else:
+        app.launch(server_name="0.0.0.0", server_port=args.port, share=True)