Update app.py

app.py

@@ -91,15 +91,20 @@ uploaded_file = st.file_uploader("Choose an image...", type="jpg")
 if uploaded_file is not None:
     # Load and display the uploaded image
     image = Image.open(uploaded_file)
-    st.image(image, caption='Uploaded Image', use_column_width=True)
     
     if st.button("Run Inference"):
         # Perform inference
         with st.spinner('Processing...'):
             generated_image = infer_image(loaded_model, image)
         
-        # Display the result
-        st.image(generated_image, caption='Generated Image', use_column_width=True)
+        # Display the original and generated images side by side
+        st.markdown("### Result")
+        col1, col2 = st.columns(2)
+
+        with col1:
+            st.image(image, caption='Original Image', use_column_width=True)
+        with col2:
+            st.image(generated_image, caption='Generated Image', use_column_width=True)
         
         # Provide a download button for the generated image
         img_byte_arr = BytesIO()