Update app.py

app.py

@@ -46,7 +46,7 @@ def load_and_prep_imgg(img : Image.Image, img_shape=224, scale=True):
     else:
         return img
         
-def predict(img) -> Tuple[Dict,float] :
+def predict(img) -> Tuple[Dict,float,float] :
 
   start_time = timer()
 
@@ -63,7 +63,7 @@ def predict(img) -> Tuple[Dict,float] :
   end_time = timer()
   pred_time = round(end_time - start_time , 4)
 
-  return pred_class , pred_time
+  return pred_class , pred_img.max() , pred_time 
 
 ### 4. Gradio app - our Gradio interface + launch command
 
@@ -79,6 +79,7 @@ example_list
 demo = gr.Interface(fn=predict ,
                     inputs=gr.Image(type='pil'),
                     outputs=[gr.Label(num_top_classes = 3 , label= 'prediction'),
+                             gr.Number(label= 'Prediction Probabilities'),
                             gr.Number(label= 'Prediction time (s)')],
                     examples = example_list,
                     title = title,