Lisandro commited on
Commit
f48ecfa
·
1 Parent(s): 5ade8f2

feat: Actualizar variables del modelo y los safetensors en app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def predict(prompt):
8
  custom_model = "lichorosario/dott_remastered_style_lora_sdxl"
9
  weight_name = "dott_style.safetensors"
10
  api_key = os.getenv('HF_API_KEY')
11
- client = Client("fffiloni/sd-xl-custom-model", api_key=api_key)
12
 
13
  # Load the model
14
  client.predict(
 
8
  custom_model = "lichorosario/dott_remastered_style_lora_sdxl"
9
  weight_name = "dott_style.safetensors"
10
  api_key = os.getenv('HF_API_KEY')
11
+ client = Client("fffiloni/sd-xl-custom-model")
12
 
13
  # Load the model
14
  client.predict(