Spaces:
Sleeping
Sleeping
Commit
•
22ea818
1
Parent(s):
cbe2334
Update app.py
Browse files
app.py
CHANGED
@@ -38,13 +38,16 @@ def predict(zip_bytes):
|
|
38 |
predicted_class = np.argmax(prediction)
|
39 |
decoded_class = label_encoder.inverse_transform([predicted_class])[0]
|
40 |
predicted_class_translation = translator(decoded_class)
|
41 |
-
predictions.append(
|
42 |
return predictions
|
43 |
|
44 |
interface = gr.Interface(
|
45 |
fn=predict,
|
46 |
inputs=gr.File(label="Выберите zip-файл (только с изображениями)"),
|
47 |
-
outputs=gr.
|
|
|
|
|
|
|
48 |
title="Нейронная сеть для классификации брака на металлопрокате"
|
49 |
)
|
50 |
|
|
|
38 |
predicted_class = np.argmax(prediction)
|
39 |
decoded_class = label_encoder.inverse_transform([predicted_class])[0]
|
40 |
predicted_class_translation = translator(decoded_class)
|
41 |
+
predictions.append({"Файл": filename, "Класс": predicted_class_translation})
|
42 |
return predictions
|
43 |
|
44 |
interface = gr.Interface(
|
45 |
fn=predict,
|
46 |
inputs=gr.File(label="Выберите zip-файл (только с изображениями)"),
|
47 |
+
outputs=gr.Table(
|
48 |
+
columns=["Файл", "Класс"],
|
49 |
+
label="Предсказания"
|
50 |
+
),
|
51 |
title="Нейронная сеть для классификации брака на металлопрокате"
|
52 |
)
|
53 |
|