Playing with different models.

README.md

@@ -8,6 +8,8 @@ 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

@@ -3,7 +3,7 @@ import json
 from transformers import pipeline
 from transformers import AutoTokenizer
 
-classifier = pipeline(task='zero-shot-classification', model='tasksource/deberta-small-long-nli')
+classifier = pipeline(task='zero-shot-classification', model='tasksource/deberta-small-long-nli', device=0)
 
 def zero_shot_classification(data_string):
     print(data_string)