Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -52,11 +52,12 @@ def classify_image(input_image):
|
|
52 |
|
53 |
# Escreva o r贸tulo de previs茫o no output_image_with_box (com a cor da caixa e estat铆sticas)
|
54 |
font = cv2.FONT_HERSHEY_SIMPLEX
|
55 |
-
|
|
|
56 |
|
57 |
# Adicione estat铆sticas (substitua com suas pr贸prias estat铆sticas)
|
58 |
stats = f"Accuracy: {prediction[0][class_index]:.2f}, Confidence: {prediction[0].max():.2f}"
|
59 |
-
cv2.putText(output_image_with_box, stats, (10, 40), font,
|
60 |
|
61 |
return output_image_with_box
|
62 |
|
|
|
52 |
|
53 |
# Escreva o r贸tulo de previs茫o no output_image_with_box (com a cor da caixa e estat铆sticas)
|
54 |
font = cv2.FONT_HERSHEY_SIMPLEX
|
55 |
+
font_scale = 0.4 # Tamanho da fonte reduzido
|
56 |
+
cv2.putText(output_image_with_box, f"Predicted Class: {predicted_class}", (10, 20), font, font_scale, (0, 255, 0), 1) # Cor verde
|
57 |
|
58 |
# Adicione estat铆sticas (substitua com suas pr贸prias estat铆sticas)
|
59 |
stats = f"Accuracy: {prediction[0][class_index]:.2f}, Confidence: {prediction[0].max():.2f}"
|
60 |
+
cv2.putText(output_image_with_box, stats, (10, 40), font, font_scale, (0, 255, 0), 1)
|
61 |
|
62 |
return output_image_with_box
|
63 |
|