Upload Preprocessing.py

Preprocessing.py

@@ -18,21 +18,23 @@ def download_and_load_pkl(url, filename):
     print(f"Attempting to download {filename} from {url}")
     try:
         response = requests.get(url, stream=True)
-        response.raise_for_status() 
-
-        with open(filename, 'wb') as f:
+        response.raise_for_status()
+        
+        temp_filepath = os.path.join('/tmp', filename)
+         
+        with open(temp_filepath, 'wb') as f:
             for chunk in response.iter_content(chunk_size=8192):
                 f.write(chunk)
         
-        obj = joblib.load(filename)
+        obj = joblib.load(temp_filepath)
         print(f"Successfully downloaded and loaded {filename}")
-        os.remove(filename) 
+        os.remove(temp_filepath) 
         return obj
     except requests.exceptions.RequestException as e:
-        print(f"Error downloading {filename} from {url}: {e}")
+        print(f"Error downloading {temp_filepath} from {url}: {e}")
         return None 
     except Exception as e:
-        print(f"Error loading {filename} after download: {e}")
+        print(f"Error loading {temp_filepath} after download: {e}")
         return None 
 
 numerical_imputer = download_and_load_pkl(NUM_IMPUTER_URL, "numerical_imputer.pkl")