amaljoy9 commited on
Commit
77e4b36
·
verified ·
1 Parent(s): 30fb055

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -35,6 +35,14 @@ from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
35
  #if not os.path.exists(hf_home):
36
  # os.makedirs(hf_home)
37
 
 
 
 
 
 
 
 
 
38
  # Run the main training script
39
  os.system('python py_new_test.py')
40
 
 
35
  #if not os.path.exists(hf_home):
36
  # os.makedirs(hf_home)
37
 
38
+ hf_home = '/tmp/hf_home'
39
+ os.environ['HF_HOME'] = hf_home
40
+
41
+ # Create the HF_HOME directory if it doesn't exist
42
+ if not os.path.exists(hf_home):
43
+ os.makedirs(hf_home)
44
+
45
+
46
  # Run the main training script
47
  os.system('python py_new_test.py')
48