Sadjad Alikhani commited on
Commit
7edd1d9
·
verified ·
1 Parent(s): 2da85bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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}...")