add

app.py

@@ -40,7 +40,7 @@ def calculate_flops_hf(model_name, input_shape, access_token, bp_factor):
     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=output_unit)
+                                                          bp_factor=bp_factor)#, output_unit=output_unit)
     print(f"Data: {data}")
 
     total_flops = data[0]['Forward+Backward FLOPs'] * 10 ** 9  # Convert GFLOPs to FLOPs