Update model_utils.py

model_utils.py

@@ -37,16 +37,16 @@ def calculate_flops_in_hugging_space(model_name: str,
     "Calculates the FLOPs and Params usage for a model init on `meta` device"
 
     try:
-        print("model_name:", model_name)
-        print("empty_model:", empty_model)
-        print("access_token:", access_token)
-        print("input_shape:", input_shape)
-        flops, macs, params, return_print  = calculate_flops_hf(model_name=model_name,
-                                                                empty_model=empty_model,
-                                                                access_token=access_token,
-                                                                input_shape=input_shape,
-                                                                return_results=True,
-                                                                output_as_string=False)
+        # print("model_name:", model_name)
+        # print("empty_model:", empty_model)
+        # print("access_token:", access_token)
+        # print("input_shape:", input_shape)
+        flops, macs, params, return_print = calculate_flops_hf(model_name=model_name,
+                                                               empty_model=empty_model,
+                                                               input_shape=input_shape,
+                                                               access_token=access_token,
+                                                               output_as_string=False,
+                                                               return_results=True)
     except Exception as e:
         print("Error info:", e)
         raise gr.Error(