first commit

app.py

@@ -70,15 +70,16 @@ if model_choice == "TinyBert Disaster Classification":
 # **3. Image Classification**
 if model_choice == "VIT Pose Classification":
     uploaded_file = st.file_uploader("Upload Image", type=["jpg", "png", "jpeg"])
-    predict = st.button("Predict Image")
+    predict = st.button("Predict Image")  
 
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
+
         col1, col2 = st.columns([1, 1])
 
         with col1:
             st.image(image, caption="Your Image", width=300)
-        
+
         with col2:
             try:
                 image_processor = AutoImageProcessor.from_pretrained(local_path, use_fast=True)
@@ -87,7 +88,8 @@ if model_choice == "VIT Pose Classification":
                 st.error("❌ Model not found. Please download the model first.")
                 pipe = None
 
-         if predict and pipe:
+        if predict and pipe:
             with st.spinner("Predicting..."):
                 output = pipe(image)
-                st.write(output)
+                st.write("### Prediction Results:")
+                st.json(output)