Update app.py

app.py

@@ -36,11 +36,14 @@ def classify_image(input_image):
     image = Image.fromarray(input_image.astype('uint8'))
     
     # Criar uma imagem com o rótulo de previsão sobreposta no centro
-    draw = ImageDraw.Draw(image)  # Mova a definição de "draw" para cá
+    draw = ImageDraw.Draw(image)
     width, height = image.size
     font = ImageFont.load_default()
     text = f'Previsão: {predicted_class_label}'
-    text_width, text_height = draw.textsize(text, font)
+    
+    # Obter o tamanho do texto
+    text_width, text_height = draw.textsize(text, font=font)
+    
     x = (width - text_width) // 2
     y = (height - text_height) // 2
     draw.text((x, y), text, fill='white', font=font)
@@ -48,6 +51,8 @@ def classify_image(input_image):
     # Converter a imagem resultante de volta para numpy
     result_image = np.array(image)
     return result_image
+# ...
+
 
 # Criar uma interface Gradio
 interface = gr.Interface(