Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
b8594c5
1
Parent(s):
7968e95
Messing with configuration.
Browse files
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
|
| 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:
|