Artiprocher commited on
Commit
3690f4e
1 Parent(s): 8601520

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -6,6 +6,17 @@ login(token=os.getenv('HF_AK'))
6
  from diffsynth import download_models
7
  download_models(["Kolors", "FLUX.1-dev"], downloading_priority=["HuggingFace", "ModelScope"])
8
 
 
 
 
 
 
 
 
 
 
 
 
9
  import gradio as gr
10
  from diffsynth import ModelManager, SDImagePipeline, SDXLImagePipeline, SD3ImagePipeline, HunyuanDiTImagePipeline, FluxImagePipeline
11
  import os, torch
 
6
  from diffsynth import download_models
7
  download_models(["Kolors", "FLUX.1-dev"], downloading_priority=["HuggingFace", "ModelScope"])
8
 
9
+ def get_file_list(path):
10
+ file_list = []
11
+ for filename in os.listdir(path):
12
+ file_path = os.path.join(path, filename)
13
+ if os.path.isdir(file_path):
14
+ file_list.extend(get_file_list(file_path))
15
+ else:
16
+ file_list.append(file_path)
17
+ return file_list
18
+ print(get_file_list("models"))
19
+
20
  import gradio as gr
21
  from diffsynth import ModelManager, SDImagePipeline, SDXLImagePipeline, SD3ImagePipeline, HunyuanDiTImagePipeline, FluxImagePipeline
22
  import os, torch