first commit

app.py

@@ -74,12 +74,20 @@ if model_choice == "VIT Pose Classification":
 
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        st.image(image, caption="Your Image", width=300)
-
-        image_processor = AutoImageProcessor.from_pretrained(local_path, use_fast=True)
-        pipe = pipeline('image-classification', model=local_path, image_processor=image_processor, device=device)
-
-        if predict:
-            with st.spinner("Predicting..."):
-                output = pipe(image)
-                st.write(output)
+        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)
+                pipe = pipeline('image-classification', model=local_path, image_processor=image_processor, device=device)
+            except OSError:
+                st.error("❌ Model not found. Please download the model first.")
+                pipe = None
+
+         if predict and pipe:
+                with st.spinner("Predicting..."):
+                    output = pipe(image)
+                    st.write(output)