Spaces:
Running
on
Zero
Running
on
Zero
Artiprocher
commited on
Commit
•
3690f4e
1
Parent(s):
8601520
Update app.py
Browse files
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
|