Frenchizer commited on
Commit
f2fd329
·
1 Parent(s): 3483f4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ labels = [
19
  "archaeology", "data science"
20
  ]
21
 
22
- context_pipeline = pipeline("zero-shot-classification", model="MoritzLaurer/deberta-v3-base-mnli-fever-anli")
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)
 
19
  "archaeology", "data science"
20
  ]
21
 
22
+ #context_pipeline = pipeline("zero-shot-classification", model="MoritzLaurer/deberta-v3-base-mnli-fever-anli")
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)