Lord-Raven commited on
Commit
b8594c5
·
1 Parent(s): 7968e95

Messing with configuration.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,8 +36,8 @@ tokenizer_name = "MoritzLaurer/deberta-v3-base-zeroshot-v2.0"
36
  # model = ORTModelForSequenceClassification.from_pretrained(model_name, export=True, provider="CUDAExecutionProvider")
37
  # tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
38
 
39
- # classifier = pipeline(task="zero-shot-classification", model=model_name, tokenizer=tokenizer_name)
40
- classifier = pipeline(task="zero-shot-classification", model=model_name, tokenizer=tokenizer_name, device="cuda:0")
41
 
42
  def classify(data_string, request: gradio.Request):
43
  if request:
 
36
  # model = ORTModelForSequenceClassification.from_pretrained(model_name, export=True, provider="CUDAExecutionProvider")
37
  # tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
38
 
39
+ # classifier = pipeline(task="zero-shot-classification", model=model_name, tokenizer=tokenizer_name, device="cuda:0")
40
+ classifier = pipeline(task="zero-shot-classification", model=model_name, tokenizer=tokenizer_name)
41
 
42
  def classify(data_string, request: gradio.Request):
43
  if request: