Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import tensorflow as tf
 import numpy as np
-from PIL import Image, ImageDraw, ImageFont
+from PIL import Image
 import cv2
 
 # Defina a camada personalizada FixedDropout
@@ -48,11 +48,11 @@ def classify_image(input_image):
         y1 = (image_height - box_size) // 2
         x2 = x1 + box_size
         y2 = y1 + box_size
-        cv2.rectangle(output_image_with_box, (x1, y1), (x2, y2), (0, 255, 0), 2)  # Exemplo de caixa verde
+        cv2.rectangle(output_image_with_box, (x1, y1), (x2, y2), (0, 255, 0), 2)  # Caixa verde
 
-    # Escreva o rótulo de previsão no output_image_with_box (com fonte menor)
+    # Escreva o rótulo de previsão no output_image_with_box (com a cor da caixa)
     font = cv2.FONT_HERSHEY_SIMPLEX
-    cv2.putText(output_image_with_box, f"Predicted Class: {predicted_class}", (10, 20), font, 0.5, (0, 0, 255), 2)
+    cv2.putText(output_image_with_box, f"Predicted Class: {predicted_class}", (10, 20), font, 0.5, (0, 255, 0), 2)  # Cor verde
 
     return output_image_with_box