Update app.py

app.py

@@ -40,7 +40,7 @@ def main():
 
         # Display the processed image in the second column
         col2.header("Processed Image")
-        col2.image(processed_image, caption="Processed Image", use_column_width=True)
+        col2.image(processed_image, caption="Segmented Image", use_column_width=True)
 
 # Function to generate an image using the PyTorch model
 def generate_image(uploaded_image):