svjack commited on
Commit
cf80bb3
·
verified ·
1 Parent(s): 4887e9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -80,7 +80,11 @@ css = """
80
  }
81
  """
82
 
83
- ld = LucidDreamer()
 
 
 
 
84
 
85
 
86
  with gr.Blocks(css=css) as demo:
@@ -336,4 +340,4 @@ with gr.Blocks(css=css) as demo:
336
 
337
 
338
  if __name__ == '__main__':
339
- demo.launch(share = True)
 
80
  }
81
  """
82
 
83
+ save_dir = "local_save"
84
+ import os
85
+ os.makedirs(save_dir, exist_ok = True)
86
+
87
+ ld = LucidDreamer(save_dir = save_dir)
88
 
89
 
90
  with gr.Blocks(css=css) as demo:
 
340
 
341
 
342
  if __name__ == '__main__':
343
+ demo.launch(share = True)