Update app.py

app.py

@@ -38,13 +38,16 @@ def predict(zip_bytes):
                     predicted_class = np.argmax(prediction)
                     decoded_class = label_encoder.inverse_transform([predicted_class])[0]
                     predicted_class_translation = translator(decoded_class)
-                    predictions.append((filename, predicted_class_translation))
+                    predictions.append({"Файл": filename, "Класс": predicted_class_translation})
     return predictions
 
 interface = gr.Interface(
   fn=predict,
   inputs=gr.File(label="Выберите zip-файл (только с изображениями)"),
-  outputs=gr.Textbox(label="Предсказанный класс"),
+  outputs=gr.Table(
+    columns=["Файл", "Класс"],
+    label="Предсказания"
+  ),
   title="Нейронная сеть для классификации брака на металлопрокате"
 )