Playing with different models.

README.md

@@ -8,8 +8,6 @@ sdk_version: 4.41.0
 app_file: app.py
 pinned: false
 license: openrail
-custom_headers:
-  access-control-allow-origin: https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -2,9 +2,18 @@ import gradio
 import json
 from transformers import pipeline
 from transformers import AutoTokenizer
+from fastapi.middleware.cors import CORSMiddleware
 
 classifier = pipeline(task='zero-shot-classification', model='tasksource/deberta-small-long-nli', device=0)
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=:"[https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win]",
+    allow_credentials=False,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 def zero_shot_classification(data_string):
     print(data_string)
     data = json.loads(data_string)

requirements.txt

@@ -1,3 +1,4 @@
+fastapi==0.88.0
 json5==0.9.10
 numpy==1.23.4
 torch==1.12.1