vkthakur88 commited on
Commit
6847cb9
1 Parent(s): ab26811

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,15 +3,14 @@ import huggingface_hub as hf_hub
3
  import gradio as gr
4
 
5
  client = hf_hub.InferenceClient(token = os.environ['HF_TOKEN'])
6
- #client.headers["x-use-cache"] = "0"
7
 
8
  def image_interface(prompt, negative_prompt, guidance_scale, steps):
9
  response = client.text_to_image(
10
  prompt = prompt,
11
  negative_prompt = negative_prompt,
12
  guidance_scale = guidance_scale,
13
- num_inference_steps = steps,
14
- model = 'stabilityai/stable-diffusion-xl-base-1.0'
15
  )
16
 
17
  return response
 
3
  import gradio as gr
4
 
5
  client = hf_hub.InferenceClient(token = os.environ['HF_TOKEN'])
6
+ client.headers["x-use-cache"] = "0"
7
 
8
  def image_interface(prompt, negative_prompt, guidance_scale, steps):
9
  response = client.text_to_image(
10
  prompt = prompt,
11
  negative_prompt = negative_prompt,
12
  guidance_scale = guidance_scale,
13
+ num_inference_steps = steps
 
14
  )
15
 
16
  return response