Update app.py

app.py

@@ -116,7 +116,7 @@ classes_dict =  {
 
 def Plot_Result(raw_plate, digits) :
     
-    fig, ax = plt.subplots(figsize=(15,10))
+    fig = plt.subplots(figsize=(15,10))
     
     #plt.figure(figsize=(15, 10))
     #plt.subplot(2, 1, 2)
@@ -147,8 +147,9 @@ def Plot_Result(raw_plate, digits) :
     #plt.axis('off')
     #plt.subplots_adjust(top=0.75)
     #plt.imshow(raw_plate)
-    ax.imshow(raw_plate)
-    ax.show()
+    #ax.imshow(raw_plate)
+    #ax.show()
+    fig = plt.imshow(raw_plate)
 
     
     
@@ -198,7 +199,8 @@ if selected_type == "Upload Image":
             
         st.markdown("------")
         
-        st.image(FINAL(img))
+        st.pyplot(FINAL(img))
+