Trying to use ONNX model.

app.py

@@ -27,8 +27,14 @@ tokenizer = AutoTokenizer.from_pretrained("typeform/mobilebert-uncased-mnli", mo
 
 classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer)
 
-def zero_shot_classification(data_string):
+def zero_shot_classification(data_string, request: gradio.Request):
     print(data_string)
+    if request:
+        print("Request headers dictionary:", request.headers)
+        print("IP address:", request.client.host)
+        print("Query parameters:", dict(request.query_params))
+    else:
+        print("No request")
     data = json.loads(data_string)
     print(data)
     results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])