Update utils/predict.py

utils/predict.py

@@ -21,10 +21,9 @@ AUTO = tf.data.AUTOTUNE
 
 def predict(image_path):
     """
-    This function is used for fetching predictions corresponding to input_dataframe.
-    It outputs another dataframe containing: 
-        1. prediction probability for each class
-        2. actual expected outcome for each entry in the input dataframe
+    This function is used for fetching predictions corresponding to input_image.
+    
+    It outputs confidence scores corresponding to each class on which the model was trained
     """
     
     test_image1 = load_img(image_path,target_size =(32,32))
@@ -48,7 +47,7 @@ def predict(image_path):
 
 
 def predict_batch(image_path):
-   
+
     test_ds = tf.data.Dataset.from_tensor_slices((x_test, y_test))
     test_ds = test_ds.batch(config.batch_size).prefetch(AUTO)
     slice = test_ds.take(1)