Spaces:
Sleeping
Sleeping
Pashtetuum
commited on
Commit
•
dbc5ea6
1
Parent(s):
775a5a1
Update app.py
Browse files
app.py
CHANGED
@@ -29,13 +29,14 @@ def predict(loaded_img):
|
|
29 |
prediction = loaded_model.predict(img_array)
|
30 |
predicted_class = np.argmax(prediction)
|
31 |
decoded_class = label_encoder.inverse_transform([predicted_class])[0]
|
32 |
-
|
|
|
33 |
return prediction
|
34 |
|
35 |
interface = gr.Interface(
|
36 |
fn=predict,
|
37 |
inputs=gr.Image(label="Загрузите изображение"),
|
38 |
-
outputs=gr.Textbox(label="
|
39 |
title="Нейронная сеть для классификации брака на металлопрокате"
|
40 |
)
|
41 |
|
|
|
29 |
prediction = loaded_model.predict(img_array)
|
30 |
predicted_class = np.argmax(prediction)
|
31 |
decoded_class = label_encoder.inverse_transform([predicted_class])[0]
|
32 |
+
accuracy = round(((np.max(prediction).item())*100),2)
|
33 |
+
prediction=f"Класс: {translator(decoded_class)} - Точность: {accuracy}%"
|
34 |
return prediction
|
35 |
|
36 |
interface = gr.Interface(
|
37 |
fn=predict,
|
38 |
inputs=gr.Image(label="Загрузите изображение"),
|
39 |
+
outputs=gr.Textbox(label="Классификация"),
|
40 |
title="Нейронная сеть для классификации брака на металлопрокате"
|
41 |
)
|
42 |
|