Spaces:
Running
on
Zero
Running
on
Zero
Update models/model.py
Browse files- models/model.py +5 -5
models/model.py
CHANGED
|
@@ -97,15 +97,15 @@ def post_model(name, model_url, version, creator):
|
|
| 97 |
modelname = model_downloader(model_url, zips_folder, unzips_folder)
|
| 98 |
|
| 99 |
if not modelname:
|
| 100 |
-
return "
|
| 101 |
|
| 102 |
model_files = get_model(unzips_folder, modelname)
|
| 103 |
|
| 104 |
if not model_files:
|
| 105 |
-
return "
|
| 106 |
|
| 107 |
if not model_files.get('pth'):
|
| 108 |
-
return "
|
| 109 |
|
| 110 |
md5_hash = calculate_md5(os.path.join(unzips_folder,model_files['pth']))
|
| 111 |
zipfile = compress(modelname, list(model_files.values()))
|
|
@@ -125,7 +125,7 @@ def post_model(name, model_url, version, creator):
|
|
| 125 |
"sr": sr,
|
| 126 |
"f0": f0
|
| 127 |
}
|
| 128 |
-
print("
|
| 129 |
# Realizar la solicitud POST
|
| 130 |
response = requests.post(api_url, files={"file": file_to_upload}, data=data)
|
| 131 |
result = response.json()
|
|
@@ -135,7 +135,7 @@ def post_model(name, model_url, version, creator):
|
|
| 135 |
result = response.json()
|
| 136 |
return json.dumps(result, indent=4)
|
| 137 |
else:
|
| 138 |
-
print("
|
| 139 |
return result
|
| 140 |
|
| 141 |
|
|
|
|
| 97 |
modelname = model_downloader(model_url, zips_folder, unzips_folder)
|
| 98 |
|
| 99 |
if not modelname:
|
| 100 |
+
return "抱歉!无法从您提供的链接下载模型,请尝试其他链接或稍后再试。"
|
| 101 |
|
| 102 |
model_files = get_model(unzips_folder, modelname)
|
| 103 |
|
| 104 |
if not model_files:
|
| 105 |
+
return "抱歉!无法找到模型文件,请检查链接对应的模型文件并稍后重试。"
|
| 106 |
|
| 107 |
if not model_files.get('pth'):
|
| 108 |
+
return "抱歉!无法找到.pth模型文件,请检查链接对应的模型文件并稍后重试。"
|
| 109 |
|
| 110 |
md5_hash = calculate_md5(os.path.join(unzips_folder,model_files['pth']))
|
| 111 |
zipfile = compress(modelname, list(model_files.values()))
|
|
|
|
| 125 |
"sr": sr,
|
| 126 |
"f0": f0
|
| 127 |
}
|
| 128 |
+
print("正在上传模型文件...")
|
| 129 |
# Realizar la solicitud POST
|
| 130 |
response = requests.post(api_url, files={"file": file_to_upload}, data=data)
|
| 131 |
result = response.json()
|
|
|
|
| 135 |
result = response.json()
|
| 136 |
return json.dumps(result, indent=4)
|
| 137 |
else:
|
| 138 |
+
print("加载模型文件时出错:", response.status_code)
|
| 139 |
return result
|
| 140 |
|
| 141 |
|