Update app.py

app.py

@@ -87,21 +87,24 @@ def prediction(image):
 st.title('License Plate Recognition 🚗')
 st.header('Upload an image of a license plate to get the License number.')
 
-# File uploader for user to upload an image
+# File uploader
 uploaded_file = st.file_uploader("Choose an image...", type="jpg")
 
+image = None
 if uploaded_file is not None:
     image = np.array(Image.open(uploaded_file))
+
+if image is not None:
     c1, c2, c3 = st.columns(3)
 
     with c1:
-        st.image(image, caption='Uploaded Image', use_column_width=True)
+        st.image(image, caption='Uploaded Image', use_container_width=True)
 
     license_plate_text, img_lp = prediction(image)
 
     with c2:
         if img_lp is not None:
-            st.image(img_lp, caption='Cropped License Plate', use_column_width=True)
+            st.image(img_lp, caption='Cropped License Plate', use_container_width=True)
         else:
             st.write('No License Plate Detected')