Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -30,6 +30,7 @@ def start_process(input_file, input_url, input2, shape0_str, shape1_str, output_
|
|
30 |
task_id = task_counter
|
31 |
input1 = input_file if input_file else input_url
|
32 |
print_log(task_id, input2, input1, "input1")
|
|
|
33 |
log = "转换过程非常慢,请耐心等待。显示文件列表不代表转换完成。如果未发生错误,转换结束会显示”任务完成“\n"
|
34 |
output_files = []
|
35 |
yield [], log
|
@@ -243,7 +244,7 @@ def start_process(input_file, input_url, input2, shape0_str, shape1_str, output_
|
|
243 |
if str(width_ratio) in input2 or width_ratio >0:
|
244 |
onnx_path = output_base + ".onnx"
|
245 |
else:
|
246 |
-
onnx_path = output_base + "-x" + str(width_ratio) + ".onnx"
|
247 |
if os.path.exists(onnx_path):
|
248 |
print_log(task_id, input2, "转换为ONNX模型", "跳过")
|
249 |
log += "跳过转换为ONNX模型\n"
|
|
|
30 |
task_id = task_counter
|
31 |
input1 = input_file if input_file else input_url
|
32 |
print_log(task_id, input2, input1, "input1")
|
33 |
+
print_log(task_id, input2, str(output_type), "output_type")
|
34 |
log = "转换过程非常慢,请耐心等待。显示文件列表不代表转换完成。如果未发生错误,转换结束会显示”任务完成“\n"
|
35 |
output_files = []
|
36 |
yield [], log
|
|
|
244 |
if str(width_ratio) in input2 or width_ratio >0:
|
245 |
onnx_path = output_base + ".onnx"
|
246 |
else:
|
247 |
+
onnx_path = output_base + "-x" + str(int(width_ratio)) + ".onnx"
|
248 |
if os.path.exists(onnx_path):
|
249 |
print_log(task_id, input2, "转换为ONNX模型", "跳过")
|
250 |
log += "跳过转换为ONNX模型\n"
|