Update app.py

app.py

@@ -37,7 +37,10 @@ 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))
+                    # Сохранение изображения в формате base64
+                    _, img_encoded = cv2.imencode('.png', img_resized)
+                    img_base64 = base64.b64encode(img_encoded).decode('utf-8')
+                    predictions.append((filename,img_base64, predicted_class_translation))
     return predictions
 
 interface = gr.Interface(