K00B404 commited on
Commit
858f710
·
verified ·
1 Parent(s): d03461e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
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"[-] Menentukan model: LoRA {'diaktifkan' if lora_add else 'tidak diaktifkan'}, model dasar: {basemodel}")
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: