From cb364fee5f47eb1526760485f251ff0ab051ae62 Mon Sep 17 00:00:00 2001
From: Szymon Bartanowicz <szymon@bartanowicz.pl>
Date: Wed, 15 May 2024 00:27:38 +0200
Subject: [PATCH] docker

---
 Dockerfile | 2 +-
 predict.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index d9a672d..8e3a8d9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@ FROM ubuntu:latest
 
 RUN apt-get update && apt-get install -y python3-pip unzip coreutils
 
-RUN pip install --no-cache-dir wheel kaggle pandas scikit-learn tensorflow
+RUN pip install --no-cache-dir wheel kaggle pandas scikit-learn tensorflow keras
 
 WORKDIR /app
 
diff --git a/predict.py b/predict.py
index 0cd688f..af76db4 100644
--- a/predict.py
+++ b/predict.py
@@ -4,6 +4,8 @@ from sklearn.preprocessing import StandardScaler, OneHotEncoder
 from sklearn.compose import ColumnTransformer
 from sklearn.pipeline import Pipeline
 from sklearn.model_selection import train_test_split
+from keras.metrics import MeanSquaredError
+
 
 loaded_model = tf.keras.models.load_model('powerlifting_model.h5')