Update app.py

app.py

@@ -42,11 +42,10 @@ elif opt == 'Upload image via link':
 if 'image' in locals():  # Check if image variable exists
     st.image(image, width=300, caption='Uploaded Image')
 
-    st.info('Worked')
-
     if st.button('Predict'):
         try:
             model = model_arc()  # Initialize your model
+            st.info('Worked')
             
             # Ensure image shape is correct and add batch dimension
             img_array = preprocess(image)  # This should return an array of shape (1, 256, 256, 3)