Hugging Face's logo

Spaces:
Ravenok
/
statosphere-backend
Running on Zero

App Files Community
Lord-Raven commited on
Commit
8217b91
·
1 Parent(s): 83433fb

Messing with configuration.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ model_name = "MoritzLaurer/deberta-v3-base-zeroshot-v2.0"
33
  file_name = "onnx/model.onnx"
34
  tokenizer_name = "MoritzLaurer/deberta-v3-base-zeroshot-v2.0"
35
 
36
- model = ORTModelForSequenceClassification.from_pretrained(model_name, file_name=file_name, export=True, provider="CUDAExecutionProvider")
37
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
38
 
39
  # model = ORTModelForSequenceClassification.from_pretrained(
 
33
  file_name = "onnx/model.onnx"
34
  tokenizer_name = "MoritzLaurer/deberta-v3-base-zeroshot-v2.0"
35
 
36
+ model = ORTModelForSequenceClassification.from_pretrained(model_name, export=True, provider="CUDAExecutionProvider")
37
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, model_max_length=512)
38
 
39
  # model = ORTModelForSequenceClassification.from_pretrained(