Playing with different models.

app.py

@@ -4,7 +4,7 @@ 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)
+classifier = pipeline(task='zero-shot-classification', model='tasksource/deberta-small-long-nli')
 
 app.add_middleware(
     CORSMiddleware,