service-internal commited on
Commit
7aaed34
·
verified ·
1 Parent(s): 38ded12

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -4
main.py CHANGED
@@ -10,10 +10,7 @@ from transformers import pipeline
10
  app = FastAPI()
11
 
12
  # Load zero-shot classification pipeline
13
- classifier = pipeline(
14
- "zero-shot-classification",
15
- model="MoritzLaurer/deberta-v3-large-zeroshot-v2.0"
16
- )
17
 
18
 
19
  # Define input schema
 
10
  app = FastAPI()
11
 
12
  # Load zero-shot classification pipeline
13
+ classifier = pipeline("zero-shot-classification", model="syedkhalid076/DeBERTa-Zero-Shot-Classification")
 
 
 
14
 
15
 
16
  # Define input schema