Update app.py

app.py

@@ -100,11 +100,11 @@ with col2:
 
 with col3:
     st.header('details')
+    colors = ["red", "yellow", "purple", "blue", "green"]
     if gibsOk:
         if clicked == -1:
             clicked = 0
         detectorForContents = pipeline("object-detection", model="sergioska/yolo-finetuned-flyers-content")
-        st.write(imagesForOpen[clicked])
         imageForDetails = Image.open(imagesForOpen[clicked])
         outputImageForDetails = np.array(imageForDetails)
         predictionsForDetails = detectorForContents(
@@ -112,14 +112,16 @@ with col3:
             candidate_labels=['title', 'price']
         )
         drawForDetails = ImageDraw.Draw(imageForDetails)
+        c = 0
         for predictionsForDetail in predictionsForDetails:
             box = predictionsForDetail["box"]
             label = predictionsForDetail["label"]
             score = predictionsForDetail["score"]
     
             xmin, ymin, xmax, ymax = box.values()
-            drawForDetails.rectangle((xmin, ymin, xmax, ymax), outline="red", width=5)
+            drawForDetails.rectangle((xmin, ymin, xmax, ymax), outline=colors[c], width=5)
             drawForDetails.text((xmin, ymin), f"{label}: {round(score,5)}", fill="white")
+            c += 1
 
         st.image(imageForDetails)
         st.write(predictionsForDetails)