Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = ["
|
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}
|