Update app.py

app.py

@@ -1,14 +1,19 @@
 import gradio as gr
 
+HEADER_NAMES = ['x-zerogpu-token', 'x-zerogpu-uuid', 'x-ip-token']
+
 def greet(name, request: gr.Request):
-    desired_headers = {k:v for k,v in request.headers.items() if k.startswith("x-")}
-    return {"greeting": "Hello " + name + "!!", "headers": desired_headers}
+    return {k:v for k,v in request.headers.items() if k in HEADER_NAMES}
 
 def headers(request: gr.Request):
-    desired_headers = {k:v for k,v in request.headers.items() if k.startswith("x-")}
-    print({"headers": desired_headers})
+    return {k:v for k,v in request.headers.items() if k in HEADER_NAMES}
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="json")
-with demo:
-    demo.load(headers, inputs=None, outputs=None)
+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()