dermanalisys-model / request.py
victormattosli's picture
Upload folder using huggingface_hub
2e248bc verified
# request.py modificado
from transformers import pipeline
print("Carregando o modelo local treinado...")
# O 'model' agora é o caminho para a pasta onde seu modelo foi salvo
pipe = pipeline("image-classification", model="./meu-modelo-ham10000-final")
# Vamos usar a mesma imagem de antes para testar
url_imagem = "https://www.msd.com.br/wp-content/uploads/sites/86/2023/10/benigno-cor.png?w=775"
print(f"Analisando a imagem: {url_imagem}")
resultado = pipe(url_imagem)
print("\nResultado da análise:")
print(resultado)