tumuyan2 commited on
Commit
fafe4e2
·
1 Parent(s): e75ced3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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"