Update app.py
Browse files
app.py
CHANGED
@@ -258,7 +258,8 @@ def generate_image(
|
|
258 |
src_inputs = []
|
259 |
use_words = []
|
260 |
cur_run_time = time.strftime("%m%d-%H%M%S")
|
261 |
-
|
|
|
262 |
temp_dir = f"{tmp_dir_root}/{session_id}/{cur_run_time}_{generate_random_string(4)}"
|
263 |
os.makedirs(temp_dir, exist_ok=True)
|
264 |
print(f"Temporary directory created: {temp_dir}")
|
|
|
258 |
src_inputs = []
|
259 |
use_words = []
|
260 |
cur_run_time = time.strftime("%m%d-%H%M%S")
|
261 |
+
processed_directory = os.environ["XVERSE_PREPROCESSED_DATA"]
|
262 |
+
tmp_dir_root = f'{processed_directory}'
|
263 |
temp_dir = f"{tmp_dir_root}/{session_id}/{cur_run_time}_{generate_random_string(4)}"
|
264 |
os.makedirs(temp_dir, exist_ok=True)
|
265 |
print(f"Temporary directory created: {temp_dir}")
|