add

app.py

@@ -43,15 +43,14 @@ def calculate_flops_hf(model_name, input_shape, access_token, bp_factor, output_
                                                           bp_factor=bp_factor, output_unit=output_unit)
     print(f"Data: {data}")
 
-    df = data['value']
-    total_flops = df['Forward+Backward FLOPs'][0] * 10 ** 9  # Convert GFLOPs to FLOPs
+    total_flops = data[0]['Forward+Backward FLOPs'] * 10 ** 9  # Convert GFLOPs to FLOPs
     threshold_flops = 10 ** 25
     meets_criteria = total_flops > threshold_flops
 
     return {
         "total_flops": total_flops,
         "meets_criteria": meets_criteria,
-        "dataframe": df,
+        "dataframe": data,
         "return_print": return_print
     }