Update app.py

app.py

@@ -116,7 +116,7 @@ classes_dict =  {
 
 def Plot_Result(raw_plate, digits) :
     
-    fig = plt.subplots(figsize=(15,10))
+    #fig = plt.subplots(figsize=(15,10))
     
     #plt.figure(figsize=(15, 10))
     #plt.subplot(2, 1, 2)
@@ -149,7 +149,16 @@ def Plot_Result(raw_plate, digits) :
     #plt.imshow(raw_plate)
     #ax.imshow(raw_plate)
     #ax.show()
-    fig = plt.imshow(raw_plate)
+    #fig = plt.imshow(raw_plate)
+
+    for img in raw_plate:
+                img_base64 = Image.fromarray(img)
+                img_base64.save(output, format="JPEG")
+
+            final_image = Image.open(output)
+            print("Opening ",final_image)
+            st.markdown("---")
+            st.image(final_image, caption='This is how your final image looks like 😉')
 
     
     
@@ -199,7 +208,7 @@ if selected_type == "Upload Image":
             
         st.markdown("------")
         
-        st.pyplot(FINAL(img))
+        FINAL(img)