From 4d9ecfb2c18e0a0e188a3846287ec94a16f7a33b Mon Sep 17 00:00:00 2001
From: rohyeeet <66509575+rohyeeet@users.noreply.github.com>
Date: Sat, 17 Oct 2020 11:38:38 +0530
Subject: [PATCH] Updated with Minor changes

---
 fashion_mnist/load_predict_cnn.cpp | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/fashion_mnist/load_predict_cnn.cpp b/fashion_mnist/load_predict_cnn.cpp
index 715fcce..e295b36 100644
--- a/fashion_mnist/load_predict_cnn.cpp
+++ b/fashion_mnist/load_predict_cnn.cpp
@@ -42,6 +42,7 @@ int ArgMax(const tensorflow::TTypes<float, 1>::Tensor& prediction);
  * @param argv[1] graph protobuf
  *
  * @return [description]
+ **
  */
 int main(int argc, char* argv[]) {
     // Initialize a tensorflow session
@@ -50,9 +51,10 @@ int main(int argc, char* argv[]) {
     if (!status.ok()) {
         std::cerr << status.ToString() << std::endl;
         return 1;
-    } else {
+    } 
+    else {
         std::cout << "Session created successfully" << std::endl;
-    }
+         }
 
     if (argc != 3)
     {
@@ -133,5 +135,5 @@ int ArgMax(const tensorflow::TTypes<float, 1>::Tensor& prediction)
         }
         std::cout << "value[" << i << "] = " << value << std::endl;
     }
-    return max_index;
+    return max_index ;
 }