Update app.py

app.py

@@ -116,11 +116,6 @@ classes_dict =  {
 
 def Plot_Result(raw_plate, digits) :
     
-    #fig = plt.subplots(figsize=(15,10))
-    
-    #plt.figure(figsize=(15, 10))
-    #plt.subplot(2, 1, 2)
-    # Read raw plate form
     raw_plate = cv2.imread(raw_plate)
     raw_plate = cv2.cvtColor(raw_plate, cv2.COLOR_BGR2RGB)
 
@@ -144,22 +139,12 @@ def Plot_Result(raw_plate, digits) :
     cv2.putText(raw_plate, digits[6], p7, cv2.FONT_HERSHEY_COMPLEX, 5, (0, 0, 0), 5, cv2.LINE_AA);
     cv2.putText(raw_plate, digits[7], p8, cv2.FONT_HERSHEY_COMPLEX, 5, (0, 0, 0), 5, cv2.LINE_AA);
 
-    #plt.axis('off')
-    #plt.subplots_adjust(top=0.75)
-    #plt.imshow(raw_plate)
-    #ax.imshow(raw_plate)
-    #ax.show()
-    #fig = plt.imshow(raw_plate)
-
-    #for img in raw_plate:
-     #   image = Image.fromarray(img)
-        #image.save('output_image.jpg')
 
     filename = 'output_image.jpg'
     cv2.imwrite(filename, raw_plate)
 
     out_img = cv2.imread("output_image.jpg")
-    st.image(out_img)
+    st.image(out_img, caption='final image ✅')
 
     
 def FINAL(img) :
@@ -199,7 +184,7 @@ if selected_type == "Upload Image":
 
     if uploaded_file is not None:
         image = Image.open(uploaded_file).convert('RGB')
-        st.image(image, width=400)
+        st.image(image, width=400, caption='input image 📷')
 
         image.save('input_image.jpg')