aliceoq commited on
Commit
809239e
·
1 Parent(s): 3a2cc70

fix download_models

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -58,6 +58,17 @@ def download_models():
58
  print("Downloaded hubert base model file successfully. File saved to ./hubert_base.pt.")
59
  else:
60
  raise Exception("Failed to download hubert base model file. Status code: " + str(response.status_code) + ".")
 
 
 
 
 
 
 
 
 
 
 
61
 
62
  download_models()
63
 
 
58
  print("Downloaded hubert base model file successfully. File saved to ./hubert_base.pt.")
59
  else:
60
  raise Exception("Failed to download hubert base model file. Status code: " + str(response.status_code) + ".")
61
+
62
+ # Download rmvpe model if not present
63
+ if not os.path.isfile('./rmvpe.pt'):
64
+ response = requests.get('https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/rmvpe.pt')
65
+
66
+ if response.status_code == 200:
67
+ with open('./rmvpe.pt', 'wb') as f:
68
+ f.write(response.content)
69
+ print("Downloaded rmvpe model file successfully. File saved to ./rmvpe.pt.")
70
+ else:
71
+ raise Exception("Failed to download rmvpe model file. Status code: " + str(response.status_code) + ".")
72
 
73
  download_models()
74