adding weight_type and gguf_filename

app.py

@@ -228,15 +228,32 @@ with demo:
                 gr.Markdown("# ✉️✨ Submit your model here!", elem_classes="markdown-text")
 
             with gr.Row():
-                with gr.Row():
-                    model_name_textbox = gr.Textbox(label="Model name")
+                with gr.Column():
+                    with gr.Row():
+                        model_name_textbox = gr.Textbox(label="Model name")
+                        
+                with gr.Column():
+                    with gr.Row():
+                        weight_type = gr.Dropdown(
+                            choices=['safetensors', 'gguf'],
+                            label="Weights type",
+                            multiselect=False,
+                            value='safetensors',
+                            interactive=True,
+                        )
+                        
+                with gr.Column():
+                    with gr.Row():
+                        gguf_filename_textbox = gr.Textbox(label="GGUF filename")
 
             submit_button = gr.Button("Submit Eval")
             submission_result = gr.Markdown()
             submit_button.click(
                 add_new_eval,
                 [
-                    model_name_textbox
+                    model_name_textbox,
+                    weight_type,
+                    gguf_filename_textbox
                 ],
                 submission_result,
             )

src/submission/submit.py

@@ -6,7 +6,7 @@ from src.display.formatting import styled_error, styled_message
 from src.envs import API, EVAL_REQUESTS_PATH, QUEUE_REPO
 
 
-def add_new_eval(model: str):
+def add_new_eval(model: str, weight_type: str, gguf_filename=None):
     user_name = ""
     model_path = model
     if "/" in model:
@@ -20,12 +20,18 @@ def add_new_eval(model: str):
         model_info = API.model_info(repo_id=model, revision='main')
     except Exception:
         return styled_error("Could not get your model information.")
+    
+    if weight_type=="safetensors":
+        if len(gguf_filename)!=0:
+            return styled_error("GGUF filename should be empty when using safetensors.")
 
     # Seems good, creating the eval
     print("Adding new eval")
 
     eval_entry = {
         "model": model,
+        "weight_type": weight_type,
+        "gguf_filename": gguf_filename,
         "status": "PENDING",
         "submitted_time": current_time,
     }