satpalsr commited on
Commit
ea7f412
·
1 Parent(s): 6089e02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import gradio as gr
2
  import os
 
3
 
4
  def get_msg(name, request: gr.Request):
5
- header_origin="no request"
6
- if request:
7
- header_origin = str(request.headers)
8
- if header_origin == str(os.environ["ORIGIN"]):
9
- return "Working"
10
  return "Hello " + name + "!" + str(header_origin)
11
 
12
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  import os
3
+ import json
4
 
5
  def get_msg(name, request: gr.Request):
6
+ header_origin = str(request.headers)
7
+ header_origin = json.loads(str(header_origin[8:-1]).replace("'", "\""))['origin']
8
+ if header_origin == str(os.environ["ORIGIN"]):
9
+ return "Working"
 
10
  return "Hello " + name + "!" + str(header_origin)
11
 
12
  with gr.Blocks() as demo: