freddyaboulton HF Staff commited on
Commit
d10c852
·
verified ·
1 Parent(s): 709bade

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,9 +2,8 @@ import gradio as gr
2
  from gradio_client import Client
3
 
4
  def set_client_for_session(request: gr.Request):
5
- x_gpu_token = request.headers['x-zerogpu-token']
6
  x_ip_token = request.headers['x-ip-token']
7
- return Client("hysts/SDXL", headers={"x-zerogpu-token": x_gpu_token, "x-ip-token": x_ip_token})
8
 
9
  def text_to_image(prompt, request: gr.Request):
10
  client = set_client_for_session(request)
 
2
  from gradio_client import Client
3
 
4
  def set_client_for_session(request: gr.Request):
 
5
  x_ip_token = request.headers['x-ip-token']
6
+ return Client("hysts/SDXL", headers={"x-ip-token": x_ip_token})
7
 
8
  def text_to_image(prompt, request: gr.Request):
9
  client = set_client_for_session(request)