Blocking requests from other origins.

app.py

@@ -18,9 +18,11 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-model_name = "xenova/mobilebert-uncased-mnli"
-model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name="onnx/model_quantized.onnx")
-tokenizer = AutoTokenizer.from_pretrained("typeform/mobilebert-uncased-mnli", model_max_length=512)
+model_name = "xenova/deberta-v3-base-tasksource-nli" # "xenova/mobilebert-uncased-mnli"
+file_name = "onnx/model_quantized.onnx"
+tokenizer_name = "sileod/deberta-v3-base-tasksource-nli" # "typeform/mobilebert-uncased-mnli"
+model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name)
+tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
 
 # file = cached_download("https://huggingface.co/" + model_name + "")
 # sess = InferenceSession(file)
@@ -28,13 +30,11 @@ 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, 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")
+        if !(request.origin in ["https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win", "https://jhuhman-statosphere-backend.hf.space"])
+            return ""
+    print(data_string)
     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'])