Update app.py

app.py

@@ -11,6 +11,7 @@ model = AutoModelForImageClassification.from_pretrained("AhmadHakami/alzheimer-i
 
 # Transformações para pré-processamento
 preprocess = transforms.Compose([
+    transforms.ToPILImage(),  # Converte para objeto PIL.Image
     transforms.Resize((224, 224)),  # Redimensiona para o tamanho de entrada do modelo (224x224)
     transforms.ToTensor(),  # Converte para tensor
     transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # Normalização
@@ -63,7 +64,7 @@ def classify_image(image):
 # Crie uma interface Gradio com instruções de uso
 iface = gr.Interface(
     fn=classify_image,
-    inputs=gr.inputs.Image(),
+    inputs=gr.inputs.Image(type="numpy", label="Carregar uma imagem"),
     outputs=gr.outputs.HTML(),  # Saída formatada com HTML
     title="Classificador de Imagem ViT para Demência com Pré-processamento",
     description="""<h1>Classificador de Imagem ViT para Demência</h1>