first commit

app.py

@@ -25,7 +25,7 @@ def download_model():
 download_model()
 
 # Carregar modelo PyTorch
-model = torch.load(MODEL_FILE)
+model = torch.load(MODEL_FILE, map_location=torch.device('cpu'))
 model.eval()
 
 # Função para gerar heatmap