Update app.py

app.py

@@ -106,11 +106,12 @@ def main():
 
         if pred_button:
             
-            outputs = int(model.predict(image))
+            outputs = model.predict(image)
             _, y_hat = outputs.max(1)
-            predicted_idx = str(y_hat.item())
-            right_column.title("Prediction")
-            right_column.write(predicted_idx)
+            # predicted_idx = str(y_hat.item())
+            # right_column.title("Prediction")
+            # right_column.write(predicted_idx)
+            right_column.write(decode_predictions(outputs, top=3)[0])
         
 
 if __name__ == '__main__':