Upload myinfer_latest.py

myinfer_latest.py

@@ -101,6 +101,7 @@ if os.path.isfile("rmvpe.pt"):
 
 
 
+
 def load_hubert():
     global hubert_model
     models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
@@ -211,7 +212,13 @@ def download_file(filename):
 
     return jsonify({'success': True, 'message': 'File downloaded successfully', 'file_path': local_file_path})
     
-
+@app.route('/list-weights', methods=['GET'])
+def list_weights():
+    directory = 'weights'
+    files = os.listdir(directory)
+    # Extract filenames without their extensions
+    filenames = [os.path.splitext(file)[0] for file in files if os.path.isfile(os.path.join(directory, file))]
+    return jsonify(filenames)
 
 @app.route("/logout")
 def logout():