From edecefe4c7cb8b3ee05090fb345de8e9aefa8a1c Mon Sep 17 00:00:00 2001
From: Zeyad Mansour <45726273+zeyad-mansour@users.noreply.github.com>
Date: Fri, 9 Dec 2022 08:04:37 -0500
Subject: [PATCH] Update main.cpp (#1167)

---
 yolov7/main.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/yolov7/main.cpp b/yolov7/main.cpp
index 4eac4ce..f92ce7b 100644
--- a/yolov7/main.cpp
+++ b/yolov7/main.cpp
@@ -1,4 +1,4 @@
-#include "config.h"
+#include "config.h"
 #include "model.h"
 #include "cuda_utils.h"
 #include "logging.h"
@@ -6,6 +6,7 @@
 #include "preprocess.h"
 #include "postprocess.h"
 #include <chrono>
+#include <fstream>
 
 using namespace nvinfer1;