Update app.py

app.py

@@ -46,7 +46,7 @@ def predict(img) ->Tuple[Dict,float]:
     probs = logits_per_image.softmax(dim=1) # we can take the softmax to get the label probabilities
     food = probs.argmax(dim=1)
     val = li[food]
-    return val
+    return val,pred_time
 
   return pred_labels_and_probs,pred_time