Update app.py

app.py

@@ -43,32 +43,20 @@ def classify_image(input_image):
     class_labels = ["Normal", "Cataract"]  # Substitua pelas suas etiquetas de classe reais
     predicted_class = class_labels[class_index]
 
-    # Crie uma imagem composta com a caixa de identificação de objeto e o rótulo de previsão
+    # Crie uma imagem composta com o rótulo de previsão
     output_image = (input_image[0] * 255).astype('uint8')
-    output_image_with_box = output_image.copy()
 
-    # Desenhe uma caixa de identificação de objeto no output_image_with_box (centralizada e maior)
-    if predicted_class == "Cataract":  # Adicione sua lógica para desenhar a caixa com base na classe
-        image_height, image_width, _ = output_image.shape
-        box_size = min(image_height, image_width) // 2
-        x1 = (image_width - box_size) // 2
-        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)  # Caixa verde
-
-    # Escreva o rótulo de previsão no output_image_with_box (com a cor da caixa)
+    # Escreva o rótulo de previsão no output_image (com a cor do texto)
     font = cv2.FONT_HERSHEY_SIMPLEX
     font_scale = 0.4  # Tamanho da fonte reduzido
 
     # Adicione a data e hora da análise
     current_time = datetime.datetime.now()
-    cv2.putText(output_image_with_box, f"Analysis Time: {current_time.strftime('%Y-%m-%d %H:%M:%S')}", (10, 40), font, font_scale, (0, 255, 0), 1)
-
-    cv2.putText(output_image_with_box, f"Predicted Class: {predicted_class}", (10, 60), font, font_scale, (0, 255, 0), 1)  # Cor verde
+    cv2.putText(output_image, f"Analysis Time: {current_time.strftime('%Y-%m-%d %H:%M:%S')}", (10, 40), font, font_scale, (0, 255, 0), 1)
 
-    return output_image_with_box
+    cv2.putText(output_image, f"Predicted Class: {predicted_class}", (10, 60), font, font_scale, (0, 255, 0), 1)  # Cor verde
 
+    return output_image
 
 # Crie uma interface Gradio
 input_interface = gr.Interface(
@@ -79,4 +67,4 @@ input_interface = gr.Interface(
 )
 
 # Inicie o aplicativo Gradio
-input_interface.launch()
+input_interface.launch()