add

app.py

@@ -41,7 +41,7 @@ def calculate_flops_hf(model_name, input_shape, access_token, bp_factor, output_
     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)
-    df = pd.DataFrame(data)
+    df = data['value']
     total_flops = df['Forward+Backward FLOPs'][0] * 10 ** 9  # Convert GFLOPs to FLOPs
     threshold_flops = 10 ** 25
     meets_criteria = total_flops > threshold_flops