Update app.py

app.py

@@ -34,11 +34,19 @@ def classify_image(input_image):
     predicted_class_label = id2label.get(str(predicted_class_id), "Desconhecido")
     # Abrir a imagem usando PIL
     image = Image.fromarray(input_image.astype('uint8'))
-    # Criar uma imagem com o rótulo de previsão sobreposta
-    from PIL import ImageDraw, ImageFont
-    draw = ImageDraw.Draw(image)
+    
+    # Calcular as coordenadas para o texto no centro da imagem
+    width, height = image.size
     font = ImageFont.load_default()
-    draw.text((10, 10), f'Previsão: {predicted_class_label}', fill='white', font=font)
+    text = f'Previsão: {predicted_class_label}'
+    text_width, text_height = draw.textsize(text, font)
+    x = (width - text_width) // 2
+    y = (height - text_height) // 2
+    
+    # Criar uma imagem com o rótulo de previsão sobreposta no centro
+    draw = ImageDraw.Draw(image)
+    draw.text((x, y), text, fill='white', font=font)
+    
     # Converter a imagem resultante de volta para numpy
     result_image = np.array(image)
     return result_image