Frenchizer commited on
Commit
d89edde
·
1 Parent(s): ed0975c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,7 +20,8 @@ labels = [
20
  "archaeology", "data science"
21
  ]
22
 
23
- context_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
24
 
25
  def detect_context(input_text):
26
  result = context_pipeline(input_text, candidate_labels=labels)
 
20
  "archaeology", "data science"
21
  ]
22
 
23
+ context_pipeline = pipeline("zero-shot-classification", model="MoritzLaurer/deberta-v3-base-mnli-fever-anli")
24
+ #context_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
25
 
26
  def detect_context(input_text):
27
  result = context_pipeline(input_text, candidate_labels=labels)