Lisandro commited on
Commit
a827413
1 Parent(s): e76f467

el modelo y los safetensors están por variables

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -9,17 +9,21 @@ api_key = os.getenv('HF_API_KEY')
9
  from gradio_client import Client
10
 
11
 
 
 
 
 
12
  def predict(prompt):
13
  client = Client("fffiloni/sd-xl-custom-model")
14
  result = client.predict(
15
- custom_model="lichorosario/dott_remastered_style_lora_sdxl",
16
  api_name="/load_model"
17
  )
18
 
19
  client = Client("fffiloni/sd-xl-custom-model")
20
  result = client.predict(
21
- custom_model="lichorosario/dott_remastered_style_lora_sdxl",
22
- weight_name="dott_style.safetensors",
23
  prompt=prompt,
24
  inf_steps=25,
25
  guidance_scale=12,
 
9
  from gradio_client import Client
10
 
11
 
12
+ custom_model="lichorosario/dott_remastered_style_lora_sdxl",
13
+ weight_name="dott_style.safetensors",
14
+
15
+
16
  def predict(prompt):
17
  client = Client("fffiloni/sd-xl-custom-model")
18
  result = client.predict(
19
+ custom_model=custom_model,
20
  api_name="/load_model"
21
  )
22
 
23
  client = Client("fffiloni/sd-xl-custom-model")
24
  result = client.predict(
25
+ custom_model=custom_model,
26
+ weight_name=weight_name,
27
  prompt=prompt,
28
  inf_steps=25,
29
  guidance_scale=12,