diff --git a/DL-prediction.py b/DL-prediction.py
index 4c8fc2f..682ddcf 100644
--- a/DL-prediction.py
+++ b/DL-prediction.py
@@ -1,9 +1,18 @@
 import pickle
+import os
 import numpy as np
 
-with open('model_with_data.pickle', 'rb') as file:
-    loaded_data = pickle.load(file)
+workspace_path = os.getenv('WORKSPACE')
+pickle_path = os.path.join(workspace_path, 'model_with_data.pickle')
 
+if os.path.exists(pickle_path):
+    with open(pickle_path, 'rb') as file:
+        loaded_data = pickle.load(file)
+else:
+    ## aby było mozna uruchomić lokalnie
+    with open('model_with_data.pickle', 'rb') as file:
+        loaded_data = pickle.load(file)
+        
 # Wczytanie modelu
 model = loaded_data[0]