Update app.py

app.py

@@ -2,8 +2,13 @@ import torch
 from torchvision import transforms
 import gradio as gr
 
-# Carregue o modelo PyTorch treinado
-model = torch.load("best.pt", map_location=torch.device('cpu'))  # Use 'cpu' se não estiver usando GPU
+# Carregue o dicionário contendo o modelo PyTorch treinado
+model_dict = torch.load("best.pt", map_location=torch.device('cpu'))  # Use 'cpu' se não estiver usando GPU
+
+# Extraia o modelo do dicionário (substitua 'model' pelo nome correto da chave, se necessário)
+model = model_dict['model']
+
+# Coloque o modelo no modo de avaliação
 model.eval()
 
 # Transformação de pré-processamento