abidlabs HF Staff commited on
Commit
98dbc07
·
verified ·
1 Parent(s): 230dc89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,7 @@ def text_to_image(prompt):
6
  from gradio.context import LocalContext
7
  request = LocalContext.request.get(None)
8
  print(f"Request available in text_to_image: {request is not None} {request}")
 
9
  client = Client("hysts/SDXL")
10
  img = client.predict(prompt, api_name="/predict")
11
  return img
@@ -15,6 +16,7 @@ def generate(prompt):
15
  from gradio.context import LocalContext
16
  request = LocalContext.request.get(None)
17
  print(f"Request available in generate: {request is not None} {request}")
 
18
  return text_to_image(prompt)
19
 
20
  with gr.Blocks() as demo:
 
6
  from gradio.context import LocalContext
7
  request = LocalContext.request.get(None)
8
  print(f"Request available in text_to_image: {request is not None} {request}")
9
+ print("x-ip", request.headers["x-ip-token"])
10
  client = Client("hysts/SDXL")
11
  img = client.predict(prompt, api_name="/predict")
12
  return img
 
16
  from gradio.context import LocalContext
17
  request = LocalContext.request.get(None)
18
  print(f"Request available in generate: {request is not None} {request}")
19
+ print("x-ip", request.headers["x-ip-token"])
20
  return text_to_image(prompt)
21
 
22
  with gr.Blocks() as demo: