Update app.py

app.py

@@ -52,7 +52,7 @@ def render_box(image, results):
         
         rect = patches.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, linewidth=2, edgecolor=color, facecolor='none')
         ax.add_patch(rect)
-        plt.text(xmin, ymin, f"{processor.id2label[label.item()]}: {score:.2f}", color=color, fontsize=12, bbox=dict(facecolor='white', alpha=0.5))
+        plt.text(xmin, ymin, f"{model.config.id2label[label.item()]}: {score:.2f}", color=color, fontsize=12, bbox=dict(facecolor='white', alpha=0.5))
     
     plt.axis('off')
     st.pyplot(plt)
@@ -72,3 +72,4 @@ else:
         image = Image.open(BytesIO(response.content))
         results = detect(image)
         render_box(image, results)
+