Santhosh Reddy commited on
Commit
4b33459
1 Parent(s): 29718f4
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -3,7 +3,7 @@ from fastapi import FastAPI
3
 
4
  app = FastAPI()
5
 
6
- classifier = pipeline("zero-shot-classification", model='cross-encoder/nli-deberta-v3-base')
7
 
8
 
9
  @app.get("/")
 
3
 
4
  app = FastAPI()
5
 
6
+ classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli')
7
 
8
 
9
  @app.get("/")