Update app.py
Browse files
app.py
CHANGED
@@ -8,11 +8,15 @@ from gradio_client import Client, handle_file
|
|
8 |
from PIL import Image
|
9 |
from huggingface_hub import login
|
10 |
from themes import IndonesiaTheme # Import custom IndonesiaTheme
|
11 |
-
|
12 |
MAX_SEED = np.iinfo(np.int32).max
|
13 |
-
HF_TOKEN = "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"
|
14 |
-
HF_TOKEN_UPSCALER = "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"
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
# Function to enable LoRA if selected
|
17 |
def enable_lora(lora_add, basemodel):
|
18 |
print(f"[-] Menentukan model: LoRA {'diaktifkan' if lora_add else 'tidak diaktifkan'}, model dasar: {basemodel}")
|
@@ -154,7 +158,7 @@ with gr.Blocks(css=css, theme=IndonesiaTheme()) as WallpaperFluxMaker:
|
|
154 |
)
|
155 |
|
156 |
lora_model_choice = gr.Dropdown(
|
157 |
-
label="🎨
|
158 |
choices=[
|
159 |
"Shakker-Labs/FLUX.1-dev-LoRA-add-details",
|
160 |
"XLabs-AI/flux-RealismLora",
|
|
|
8 |
from PIL import Image
|
9 |
from huggingface_hub import login
|
10 |
from themes import IndonesiaTheme # Import custom IndonesiaTheme
|
11 |
+
from loras import loras
|
12 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
|
|
13 |
|
14 |
+
HF_TOKEN = os.getenv('HF_TOKEN_UPSCALER')
|
15 |
+
HF_TOKEN_UPSCALER = os.getenv('HF_TOKEN_UPSCALER')
|
16 |
+
|
17 |
+
for lora in loras:
|
18 |
+
print(lora.repo)
|
19 |
+
|
20 |
# Function to enable LoRA if selected
|
21 |
def enable_lora(lora_add, basemodel):
|
22 |
print(f"[-] Menentukan model: LoRA {'diaktifkan' if lora_add else 'tidak diaktifkan'}, model dasar: {basemodel}")
|
|
|
158 |
)
|
159 |
|
160 |
lora_model_choice = gr.Dropdown(
|
161 |
+
label="🎨 select LoRA",
|
162 |
choices=[
|
163 |
"Shakker-Labs/FLUX.1-dev-LoRA-add-details",
|
164 |
"XLabs-AI/flux-RealismLora",
|