enable benchmarking.

main.py

@@ -71,12 +71,12 @@ def run(hf_token: str) -> str:
     tensorrt_path = "trt_resnet50_keras"
     convert_to_trt(saved_model_path, tensorrt_path)
 
-    # tf_throughput = benchmark(model)
-    # tf_trt_throughput = benchmark(tensorrt_path)
+    tf_throughput = benchmark(model)
+    tf_trt_throughput = benchmark(tensorrt_path)
 
-    # benchmark_str = post_optimization([tf_throughput, tf_trt_throughput])
-    # benchmark_str += "\n"
-    benchmark_str = push_to_hub(hf_token, tensorrt_path)
+    benchmark_str = post_optimization([tf_throughput, tf_trt_throughput])
+    benchmark_str += "\n"
+    benchmark_str += push_to_hub(hf_token, tensorrt_path)
 
     return benchmark_str