Update app.py

app.py

@@ -69,8 +69,7 @@ if uploaded_image is not None and uploaded_query is not None and start_button:
         plt.imshow(img[:,:,::-1])
         plt.text(
               cx ,
-              cy ,
-              #class_name+str(round(score.tolist(),2)) ,
+              cy , #class_name+str(round(score.tolist(),2)) ,
               ha="left",
               va="top",
               color="red",
@@ -80,10 +79,8 @@ if uploaded_image is not None and uploaded_query is not None and start_button:
                   "boxstyle": "square,pad=.3"
               })
       
-        #output_image = fig.canvas.tostring_rgb()
-        plt.close()
-        output_image = img[:,:,::-1]
-        st.image(output_image, caption='Predicted Image', use_column_width=True)
+    output_image = img[:,:,::-1]
+    st.image(output_image, caption='Predicted Image', use_column_width=True)
 
 else:
     st.write('Please upload an image and a query image.')