adding Inference time feature

app.py

@@ -1,9 +1,11 @@
+import time
 import gradio as gr
 import tensorflow as tf
 import matplotlib.pyplot as plt
 
 def predict(input_img):
 
+    start = time.time()
     # Load the saved Keras model
     model = tf.keras.models.load_model("VGG19.h5")
     # Preprocess the image
@@ -29,14 +31,16 @@ def predict(input_img):
               f'{item[1]} % {class_names[1]}\n'
               f'{item[2]} % {class_names[2]}\n')
     plt.imshow(img_0/255)
-    return input_img, plt
+    end = time.time()
+    inf_time = end - start
+    return plt, inf_time
 
     
 output = gr.Plot()
 gradio_app = gr.Interface(
     predict,
     inputs=gr.Image(label="table type", sources=['upload', 'webcam'], type="pil"),
-    outputs=[gr.Image(label="Processed Image"), output],
+    outputs=[output, gr.Textbox(label="Inference time")],
     title="Table-type Classification"
 )