gsbm commited on
Commit
5016deb
·
verified ·
1 Parent(s): 5298720

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -45,7 +45,7 @@ def generate(
45
  num_inference_steps_refiner: int = 25,
46
  apply_refiner: bool = False,
47
  ):
48
- client = Client("hysts/SDXL")
49
  result = client.predict(
50
  prompt=prompt,
51
  negative_prompt=negative_prompt,
@@ -63,7 +63,6 @@ def generate(
63
  num_inference_steps_refiner=num_inference_steps_refiner,
64
  apply_refiner=apply_refiner,
65
  api_name="/run",
66
- hf_token=os.environ['HF_TOKEN']
67
  )
68
  image = pixelate(result, 16)
69
  return image
 
45
  num_inference_steps_refiner: int = 25,
46
  apply_refiner: bool = False,
47
  ):
48
+ client = Client("hysts/SDXL", hf_token=os.environ['HF_TOKEN'])
49
  result = client.predict(
50
  prompt=prompt,
51
  negative_prompt=negative_prompt,
 
63
  num_inference_steps_refiner=num_inference_steps_refiner,
64
  apply_refiner=apply_refiner,
65
  api_name="/run",
 
66
  )
67
  image = pixelate(result, 16)
68
  return image