add

app.py

@@ -33,14 +33,14 @@ def calculate_flops_gpu(gpu_hours, power_consumption_w, flops_per_gpu_s):
     }
 
 
-def calculate_flops_hf(model_name, input_shape, access_token, bp_factor, output_unit):
+def calculate_flops_hf(model_name, input_shape, access_token, bp_factor):
     """
     Calculate FLOPs using Hugging Face model information.
     """
     model = get_mode_from_hf(model_name=model_name, library="auto", access_token=access_token)
     data, return_print = calculate_flops_in_hugging_space(model_name=model_name, empty_model=model,
                                                           access_token=access_token, input_shape=input_shape,
-                                                          bp_factor=bp_factor)#, output_unit=output_unit)
+                                                          bp_factor=bp_factor,output_unit="")#, output_unit=output_unit)
     print(f"Data: {data}")
 
     total_flops = data[0]['Forward+Backward FLOPs'] * 10 ** 9  # Convert GFLOPs to FLOPs