diff --git a/1/model.py b/1/model.py
index 15ad737..023c020 100644
--- a/1/model.py
+++ b/1/model.py
@@ -1,3 +1,4 @@
+import os
 import torch
 import json
 import numpy as np