Update app.py

app.py

@@ -42,6 +42,8 @@ if uploaded_file is not None:
     tensor,tensor.shape
     custom_pred = model.forward(tensor).detach().numpy() # Forward is the python method defined inside the resnet.
     custom_pred
+    st.write(f'Predicted: {id2class[np.argmax(custom_pred)]}')
+    
   elif ".csv" in uploaded_file.name: #csv check
     dataframe = pd.read_csv(uploaded_file)
     st.write(dataframe)