Lord-Raven commited on
Commit
02f2915
·
1 Parent(s): afbe2ec

Messing with configuration.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,13 +42,13 @@ session_options.log_severity_level = 0
42
 
43
  print(f"ORTModelForSequenceClassification.from_pretrained")
44
  model = ORTModelForSequenceClassification.from_pretrained(
45
- "philschmid/tiny-bert-sst2-distilled",
46
  export=True,
47
  provider="CUDAExecutionProvider",
48
  session_options=session_options
49
  )
50
  print(f"AutoTokenizer.from_pretrained")
51
- tokenizer = AutoTokenizer.from_pretrained("philschmid/tiny-bert-sst2-distilled")
52
 
53
  # classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer, device="cuda:0")
54
 
 
42
 
43
  print(f"ORTModelForSequenceClassification.from_pretrained")
44
  model = ORTModelForSequenceClassification.from_pretrained(
45
+ "distilbert-base-uncased-finetuned-sst-2-english",
46
  export=True,
47
  provider="CUDAExecutionProvider",
48
  session_options=session_options
49
  )
50
  print(f"AutoTokenizer.from_pretrained")
51
+ tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
52
 
53
  # classifier = pipeline(task="zero-shot-classification", model=model, tokenizer=tokenizer, device="cuda:0")
54