fix tabs

app.py

@@ -66,25 +66,24 @@ st.title('Pose Estimation App')
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 st.caption(f'Using device: {device}')
 
-upload_tab, camera_tab = st.tabs(["Upload Photo", "Webcam Capture"])
+mode = st.radio('Select mode:', ['Upload an Image', 'Webcam Capture'])
 
-with upload_tab:
-    uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-
-with camera_tab:
-    img_file_buffer = st.camera_input("Take a picture")
+if mode == 'Upload an Image':
+    img_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+elif mode == 'Webcam Capture':
+    img_file = st.camera_input("Take a picture")
 
 img = None
-if img_file_buffer is not None:
-    img = Image.open(img_file_buffer)
-if uploaded_file is not None:
-    img = Image.open(uploaded_file)
+if img_file is not None:
+    img = Image.open(img_file)
 
+st.divider()
 if img is not None:
     # predict
     with st.spinner('Predicting...'):
         model = load_model()
         pred = model(img)[0]
+    st.markdown('**Results:**')
     keypoints = infer.get_keypoints(pred)
     if keypoints is not None:
         img = draw_output(img, keypoints)