Update app.py

app.py

@@ -244,10 +244,14 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
         except Exception as e:
             log += f"执行命令: {command} 失败，错误信息: {str(e)}\n"
             
-        # 检查文件是否存在
-        bin_file = os.path.join(output_folder, input2 + ".ncnn.bin")
-        param_file = os.path.join(output_folder, input2 + ".ncnn.param")
-        if os.path.exists(bin_file) and os.path.exists(param_file):
+        # 查找 output_folder 目录下以 .ncnn.bin 和 .ncnn.param 结尾的文件
+        bin_files = [f for f in os.listdir(output_folder) if f.endswith('.ncnn.bin')]
+        param_files = [f for f in os.listdir(output_folder) if f.endswith('.ncnn.param')]
+        
+
+        if bin_files and param_files:
+            param_file = os.path.join(output_folder, param_files[0])
+            bin_file = os.path.join(output_folder, bin_files[0])
             import zipfile
             # 压缩包名称
             zip_file_name = os.path.join(output_folder, f"models-{input2}.zip")
@@ -265,7 +269,11 @@ def start_process(input1, input2, shape0_str, shape1_str, input_suffix=".pth"):
             log += f"已创建压缩包: {zip_file_name}\n"
             print_log(task_id, input2, "创建压缩包"+zip_file_name, "完成")
             yield [], log
-
+        else:
+            log += f"未找到 ncnn 文件\n"
+            print_log(task_id, input2, "查找 ncnn 文件", "失败")
+            yield [], log
+            
         output_files = [os.path.join(output_folder, f) for f in os.listdir(output_folder) if os.path.isfile(os.path.join(output_folder, f))]
         log += f"任务完成\n"
         print_log(task_id, input2, "执行命令", "完成")