diff --git a/Biblioteka_DL/dllib-mlflow.py b/Biblioteka_DL/dllib-mlflow.py
index 80a74b1..4995166 100644
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@@ -31,10 +31,10 @@ class Model(nn.Module):
         self.layer3 = nn.Linear(40, 3)
 
     def forward(self, x):
-        x = F.relu(self.layer1(x))
-        x = F.relu(self.layer2(x))
-        x = F.softmax(self.layer3(x))  # To check with the loss function
-        return x
+        x = F.relu(self.layer1(x.float()))
+        x = F.relu(self.layer2(x.float()))
+        x = F.softmax(self.layer3(x.float()))  # To check with the loss function
+        return x.float()
 
 # funkcja usuwająca wiersze zawierające platformę "Stadia"
 def delete_stadia(games):