Update models/model.py

models/model.py

@@ -106,10 +106,18 @@ def post_model(name, model_url, version, creator):
 
     if not model_files.get('pth'):
         return "抱歉！无法找到.pth模型文件，请检查链接对应的模型文件并稍后重试。"
-    
+    else:
+        print("已找到.pth模型文件")
+      
+    if not model_files.get('index'):
+        return "抱歉！无法找到.index模型文件，请检查链接对应的模型文件并稍后重试。"
+    else:
+        print("已找到.index模型文件")
+        
     md5_hash = calculate_md5(os.path.join(unzips_folder,model_files['pth']))
     zipfile = compress(modelname, list(model_files.values()))
     
+    print(f"已打包模型文件：{model_files.values()}")
     a = get_info(model_files.get('pth'))
     file_to_upload = open(zipfile, "rb")
     info = a.get("info", "None"),