Update app.py

app.py

@@ -5,9 +5,8 @@ def predict(text, request: gr.Request):
     client_ip = request.client.host
     local_ip = socket.gethostbyname(socket.gethostbyname(""))
     print(request)
-    headers = request.headers
-    if headers and 'x-forwarded-for' in headers:
-        x_forwarded_for = headers['x-forwarded-for']
+    x_forwarded_for = request.headers.get('x-forwarded-for')
+    if x_forwarded_for:
         client_ip = x_forwarded_for.split(' ')[0] if x_forwarded_for else ""
 
     print("client_ip", client_ip, text, "\n\n")