Spaces:
Running
Running
Sadjad Alikhani
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -103,6 +103,10 @@ def process_p_file(uploaded_file, percentage_idx, complexity_idx):
|
|
103 |
print(f"sys.path: {sys.path}")
|
104 |
#sys.path.append('./')
|
105 |
|
|
|
|
|
|
|
|
|
106 |
from lwm_model import LWM
|
107 |
device = 'cpu'
|
108 |
print(f"Loading the LWM model on {device}...")
|
|
|
103 |
print(f"sys.path: {sys.path}")
|
104 |
#sys.path.append('./')
|
105 |
|
106 |
+
lwm_model_dir = os.path.join(os.getcwd(), 'LWM')
|
107 |
+
if lwm_model_dir not in sys.path:
|
108 |
+
sys.path.append(lwm_model_dir)
|
109 |
+
|
110 |
from lwm_model import LWM
|
111 |
device = 'cpu'
|
112 |
print(f"Loading the LWM model on {device}...")
|