Spaces:
Running
Running
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 |
|