Update app.py

removing precision for now

app.py

@@ -35,22 +35,6 @@ class WeightType(Enum):
     Original = ModelDetails("Original")
     Delta = ModelDetails("Delta")
 
-class Precision(Enum):
-    float16 = ModelDetails("float16")
-    bfloat16 = ModelDetails("bfloat16")
-    float32 = ModelDetails("float32")
-    bfloat32 = ModelDetails("bfloat32")
-    Unknown = ModelDetails("?")
-
-    def from_str(precision):
-        if precision in ["torch.float16", "float16"]:
-            return Precision.float16
-        if precision in ["torch.bfloat16", "bfloat16"]:
-            return Precision.bfloat16
-        if precision in ["torch.bfloat32", "bfloat32"]:
-            return Precision.bfloat32
-        if precision in ["torch.float32", "float32"]:
-            return Precision.float32
             
 class ModelType(Enum):
     PT = ModelDetails(name="pretrained", symbol="🟢")
@@ -83,26 +67,21 @@ def start_compute_space():
     return f"Okay! {COMPUTE_SPACE} should be running now!"
 
 
-def get_model_size(model_info: ModelInfo, precision: str):
+def get_model_size(model_info: ModelInfo):
     """Gets the model size from the configuration, or the model name if the configuration does not contain the information."""
     try:
         model_size = round(model_info.safetensors["total"] / 1e9, 3)
     except (AttributeError, TypeError):
         return 0  # Unknown model sizes are indicated as 0, see NUMERIC_INTERVALS in app.py
-    
-    size_factor = 8 if (precision == "GPTQ" or "gptq" in model_info.modelId.lower()) else 1
-    model_size = size_factor * model_size
     return model_size
 
 
 def add_new_eval(
     repo_id: str,
-    precision: str,
     task: str,
 ):
     model_owner = repo_id.split("/")[0]
-    model_name = repo_id.split("/")[1]
-    precision = precision.split(" ")[0]
+    model_name = repo_id.split("/")[1]  
     
     current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
 
@@ -114,7 +93,7 @@ def add_new_eval(
         return
     #    return styled_error("Could not get your model information. Please fill it up properly.")
 
-    model_size = get_model_size(model_info=model_info, precision=precision)
+    model_size = get_model_size(model_info=model_info)
     
     print("Adding request")
 
@@ -122,7 +101,7 @@ def add_new_eval(
     
     request_dict = {
         "model": repo_id,
-        "precision": precision,
+        "precision": "N/A",
         "status": "PENDING",
         "submitted_time": pd.to_datetime(current_time),
         "task": task,
@@ -158,18 +137,12 @@ with gr.Blocks() as demo:
             task = gr.Dropdown(
                 choices=tasks,
                 label="Choose a benchmark task",
+                value = 'Text Generation',
                 multiselect=False,
                 interactive=True,
             )
         with gr.Column():
             model_name_textbox = gr.Textbox(label="Model name")
-            precision = gr.Dropdown(
-                choices=[i.value.name for i in Precision if i != Precision.Unknown],
-                label="Precision",
-                multiselect=False,
-                value="float16",
-                interactive=True,
-            )
 
     with gr.Row():
         with gr.Column():
@@ -179,7 +152,6 @@ with gr.Blocks() as demo:
                 fn=add_new_eval,
                 inputs=[
                     model_name_textbox,
-                    precision,
                     task,
                 ],
                 outputs=submission_result,