Pashtetuum commited on
Commit
dbc5ea6
1 Parent(s): 775a5a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- prediction=translator(decoded_class)
 
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