Update app.py

app.py

@@ -56,7 +56,7 @@ def ui():
         pil_image = PIL.Image.fromarray(image)
         image = preprocess(pil_image).unsqueeze(0).to(device)
 
-        option = st.selectbox('Please select the Model',('PreTrained Model'))
+        option = st.selectbox('Please select the Model',('PreTrained Model','Trained Model'))
 
         if option=='PreTrained Model':
             with torch.no_grad():