MJobe commited on
Commit
02494dd
1 Parent(s): 0e96068

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -38,8 +38,7 @@ nlp_qa_v3 = pipeline("question-answering", model="deepset/roberta-base-squad2")
38
  nlp_classification = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
39
  nlp_classification_v2 = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
40
  nlp_speech_to_text = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h")
41
- nlp_sequence_classification = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-1")
42
-
43
  description = """
44
  ## Image-based Document QA
45
  This API performs document question answering using a LayoutLMv2-based model.
 
38
  nlp_classification = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
39
  nlp_classification_v2 = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
40
  nlp_speech_to_text = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h")
41
+ nlp_sequence_classification = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
42
  description = """
43
  ## Image-based Document QA
44
  This API performs document question answering using a LayoutLMv2-based model.