mitesh001 commited on
Commit
480eb72
·
1 Parent(s): 6c1d0ee

Update space and updated models for high accuracy

Browse files
Files changed (1) hide show
  1. main.py +6 -3
main.py CHANGED
@@ -13,9 +13,12 @@ import time
13
  app = FastAPI()
14
 
15
  # Load classification and summarization models
16
- classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
17
- summarizer_tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
18
- summarizer_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
 
 
 
19
 
20
  # Load Indic NER (or any general one)
21
  tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
 
13
  app = FastAPI()
14
 
15
  # Load classification and summarization models
16
+ classifier = pipeline("zero-shot-classification", model="joeddav/xlm-roberta-large-xnli")
17
+ summarizer_tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
18
+ summarizer_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
19
+ # classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
20
+ # summarizer_tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
21
+ # summarizer_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
22
 
23
  # Load Indic NER (or any general one)
24
  tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")