Update app.py

app.py

@@ -126,29 +126,27 @@ def main():
             # Display uploaded image
             image = Image.open(uploaded_file)
             st.image(image, caption="Uploaded Image", use_column_width=True)
-            
-            # Add predict button
-            if st.button("Predict"):
-                with st.spinner('Analyzing image...'):
-                    try:
-                        # Load model
-                        model = load_model()
-                        
-                        # Preprocess image
-                        processed_image = preprocess_image(image)
-                        
-                        # Make prediction
-                        prediction = model.predict(np.expand_dims(processed_image, axis=0))
-                        class_names = ['10', '6.5', '7', '7.5', '8', '8.5', '9', '9.2', '9.5', '9.7']
-                        
-                        # Get top 5 predictions
-                        top_predictions = get_top_predictions(prediction, class_names)
-                        
-                        # Store in session state
-                        st.session_state.predictions = top_predictions
-                        
-                    except Exception as e:
-                        st.error(f"Error during prediction: {str(e)}")
+
+            with st.spinner('Analyzing image...'):
+                try:
+                    # Load model
+                    model = load_model()
+                    
+                    # Preprocess image
+                    processed_image = preprocess_image(image)
+                    
+                    # Make prediction
+                    prediction = model.predict(np.expand_dims(processed_image, axis=0))
+                    class_names = ['10', '6.5', '7', '7.5', '8', '8.5', '9', '9.2', '9.5', '9.7']
+                    
+                    # Get top 5 predictions
+                    top_predictions = get_top_predictions(prediction, class_names)
+                    
+                    # Store in session state
+                    st.session_state.predictions = top_predictions
+                    
+                except Exception as e:
+                    st.error(f"Error during prediction: {str(e)}")
     
     with col2:
         st.subheader("Prediction Results")