test-headers / app.py
freddyaboulton's picture
Update app.py
a599314 verified
raw
history blame contribute delete
631 Bytes
import gradio as gr
HEADER_NAMES = ['x-zerogpu-token', 'x-zerogpu-uuid', 'x-ip-token']
def greet(name, request: gr.Request):
return {k:v for k,v in request.headers.items() if k in HEADER_NAMES}
def headers(request: gr.Request):
return {k:v for k,v in request.headers.items() if k in HEADER_NAMES}
iface = gr.Interface(fn=greet, inputs="text", outputs="json")
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
iface.render()
with gr.Column():
load_headers = gr.Json(label="Headers on Page Load")
demo.load(headers, inputs=None, outputs=load_headers)
demo.launch()