DHEIVER commited on
Commit
c41ab68
1 Parent(s): 7f4a879

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def classificar_imagem(imagem):
17
  confianca = logits.softmax(dim=1).max().item()
18
 
19
  # Mapear a classe prevista para os nomes corretos
20
- nomes_classes = ["meningioma_tumor", "glioma_tumor", "no_tumor", "pituitary_tumor"]
21
  classe_prevista_texto = nomes_classes[classe_prevista]
22
 
23
  return {"previsao": classe_prevista_texto, "confianca": confianca}
 
17
  confianca = logits.softmax(dim=1).max().item()
18
 
19
  # Mapear a classe prevista para os nomes corretos
20
+ nomes_classes = ["glioma_tumor", "meningioma_tumor", "no_tumor", "pituitary_tumor"]
21
  classe_prevista_texto = nomes_classes[classe_prevista]
22
 
23
  return {"previsao": classe_prevista_texto, "confianca": confianca}