sileod commited on
Commit
7400288
·
verified ·
1 Parent(s): 6b5f14e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -5,6 +5,9 @@ from transformers import pipeline
5
  zero_shot_classifier = pipeline("zero-shot-classification", model="tasksource/ModernBERT-base-nli")
6
  nli_classifier = pipeline("text-classification", model="tasksource/ModernBERT-base-nli")
7
 
 
 
 
8
  def process_input(text_input, labels_or_premise, mode):
9
  if mode == "Zero-Shot Classification":
10
  labels = [label.strip() for label in labels_or_premise.split(',')]
 
5
  zero_shot_classifier = pipeline("zero-shot-classification", model="tasksource/ModernBERT-base-nli")
6
  nli_classifier = pipeline("text-classification", model="tasksource/ModernBERT-base-nli")
7
 
8
+ if False:
9
+ gr.load("models/answerdotai/ModernBERT-base").launch()
10
+
11
  def process_input(text_input, labels_or_premise, mode):
12
  if mode == "Zero-Shot Classification":
13
  labels = [label.strip() for label in labels_or_premise.split(',')]