add upload

app.py

@@ -429,6 +429,8 @@ with demo:
                         value=None,
                         interactive=True,
                     )
+                    runsh = gr.File(label="上传 run.sh 文件", file_types=[".sh"])
+                    adapter = gr.File(label="上传 model_adapter.py 文件", file_types=[".py"])
 
                 with gr.Column():
                     private = gr.Checkbox(False, label="Private", visible=not IS_PUBLIC)
@@ -464,6 +466,8 @@ with demo:
                     private,
                     weight_type,
                     model_type,
+                    runsh,
+                    adapter,
                 ],
                 submission_result,
             )

src/submission/submit.py

@@ -26,6 +26,8 @@ def add_new_eval(
     private: str,
     weight_type: str,
     model_type: str,
+    runsh,
+    adapter
 ):
     global REQUESTED_MODELS
     global USERS_TO_SUBMISSION_DATES
@@ -103,7 +105,16 @@ def add_new_eval(
 
     # Seems good, creating the eval
     print("Adding new eval")
-
+    max_size = 5 * 1024 * 1024  # 5MB
+    if os.path.getsize(runsh.name) > max_size:
+        return "错误：文件大小不能超过 5MB！"
+    if os.path.getsize(adapter.name) > max_size:
+        return "错误：文件大小不能超过 5MB！"
+
+    with open(runsh.name, "r") as f:
+        runsh = f.read()
+    with open(adapter.name, "r") as f:
+        adapter = f.read()
     eval_entry = {
         "model": model,
         "model_api_url": model_api_url,
@@ -121,6 +132,8 @@ def add_new_eval(
         "params": model_size,
         #"license": license,
         "private": False,
+        "runsh": runsh,
+        "adapter": adapter,
     }
 
     supplementary_info = {