lionelgarnier commited on
Commit
d4c4c25
·
1 Parent(s): 3103d69
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -18,19 +18,19 @@ from trellis.utils import render_utils, postprocessing_utils
18
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
21
- os.makedirs(TMP_DIR, exist_ok=True)
22
 
23
 
24
- def start_session(req: gr.Request):
25
- gr.Warning('start start session')
26
- user_dir = os.path.join(TMP_DIR, str(req.session_hash))
27
- os.makedirs(user_dir, exist_ok=True)
28
- gr.Warning('end start session')
29
 
30
 
31
- def end_session(req: gr.Request):
32
- user_dir = os.path.join(TMP_DIR, str(req.session_hash))
33
- shutil.rmtree(user_dir)
34
 
35
 
36
  # def preprocess_image(image: Image.Image) -> Image.Image:
@@ -245,12 +245,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
245
  download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
246
  download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)
247
 
248
- output_buf = gr.State()
249
 
250
 
251
  # Handlers
252
- demo.load(start_session)
253
- demo.unload(end_session)
254
 
255
 
256
  # image_prompt.upload(
 
18
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
21
+ # os.makedirs(TMP_DIR, exist_ok=True)
22
 
23
 
24
+ # def start_session(req: gr.Request):
25
+ # gr.Warning('start start session')
26
+ # user_dir = os.path.join(TMP_DIR, str(req.session_hash))
27
+ # os.makedirs(user_dir, exist_ok=True)
28
+ # gr.Warning('end start session')
29
 
30
 
31
+ # def end_session(req: gr.Request):
32
+ # user_dir = os.path.join(TMP_DIR, str(req.session_hash))
33
+ # shutil.rmtree(user_dir)
34
 
35
 
36
  # def preprocess_image(image: Image.Image) -> Image.Image:
 
245
  download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
246
  download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)
247
 
248
+ # output_buf = gr.State()
249
 
250
 
251
  # Handlers
252
+ # demo.load(start_session)
253
+ # demo.unload(end_session)
254
 
255
 
256
  # image_prompt.upload(