put the digits in the sidebar

app.py

@@ -7,8 +7,9 @@ import torch
 from torchvision.datasets import MNIST
 from torchvision.transforms import ToTensor
 import numpy as np
-st.title("Streamlit GUI")
-
+st.title("MNIST classifier")
+st.markdown("This is a simple MNIST classifier using a simple neural network")
+st.markdown("select a digit from the sidebar and the classifier will give you the probability of each digit")
 classifier=AutoModel.from_pretrained("hikmatfarhat/MNIST_Classifier",trust_remote_code=True)
 softmax=torch.nn.Softmax(dim=1)
 np.set_printoptions(precision=3,suppress=True)
@@ -16,7 +17,8 @@ imgs=[]
 
 for i in range(15):
     imgs.append(Image.open(f"img{i}.png"))
-image=image_select("select a digit",imgs,use_container_width=False)
+with st.sidebar:
+    image=image_select("select a digit",imgs,use_container_width=False)
 image=ToTensor()(image)
 output=classifier(image)
 output=torch.squeeze(softmax(output))