Update app.py
Browse files
app.py
CHANGED
@@ -13,11 +13,19 @@ 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"[-]
|
19 |
return basemodel if not lora_add else lora_add
|
20 |
-
|
21 |
# Function to generate image
|
22 |
async def generate_image(prompt, model, lora_word, width, height, scales, steps, seed):
|
23 |
try:
|
|
|
13 |
HF_TOKEN = "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"
|
14 |
HF_TOKEN_UPSCALER = "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"
|
15 |
|
16 |
+
HF_TOKEN = os.getenv('HF_TOKEN_UPSCALER')
|
17 |
+
HF_TOKEN_UPSCALER = os.getenv('HF_TOKEN_UPSCALER')
|
18 |
+
qwen_client = Client("K00B404/HugChatWrap",hf_token=HF_TOKEN)
|
19 |
+
loaded_loras=[]
|
20 |
+
for lora in loras:
|
21 |
+
print(lora.get('repo'))
|
22 |
+
loaded_loras.append(lora.get('repo'))
|
23 |
+
|
24 |
# Function to enable LoRA if selected
|
25 |
def enable_lora(lora_add, basemodel):
|
26 |
+
print(f"[-] Determining model: LoRA {'enabled' if lora_add else 'disabled'}, base model: {basemodel}")
|
27 |
return basemodel if not lora_add else lora_add
|
28 |
+
|
29 |
# Function to generate image
|
30 |
async def generate_image(prompt, model, lora_word, width, height, scales, steps, seed):
|
31 |
try:
|