Update app.py
Browse files
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 |
|