Update app.py

app.py

@@ -78,8 +78,9 @@ def draw_bbox(image, bbox):
     return image
 
 
+st.title("AI INTEGRATION FOR CHEST X-RAY")
 
-# st.title("Chest X-ray Disease Detection")
+st.header("Chest X-ray Disease Detection")
 
 st.write("Upload a chest X-ray image and click on 'Detect' to find out if there's any disease.")
 
@@ -91,7 +92,7 @@ if uploaded_file is not None:
     file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
     image = cv2.imdecode(file_bytes, 1)
 
-    st.image(image, caption='Uploaded Image.', use_column_width=False, width=320, height=320)
+    st.image(image, caption='Uploaded Image.', use_column_width=True)
 
     if st.button('Detect'):
         st.write("Processing...")
@@ -119,7 +120,7 @@ if uploaded_file is not None:
             st.write(f"Prediction Confidence: {pred_label_confidence:.2f}")
 
             output_image = draw_bbox(image.copy(), pred_bbox)
-            st.image(output_image, caption='Detected Image.', use_column_width=False, width=320, height=320)
+            st.image(output_image, caption='Detected Image.', use_column_width=True)
 
 @st.cache_resource
 def load_gradcam_model():