oliv32 commited on
Commit
520f023
·
1 Parent(s): 7e93ddb

Fix: Corrected model ID to oliv32/bert-fine-tuned.

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -8,11 +8,11 @@ from transformers import (
8
  AutoTokenizer,
9
  )
10
 
11
- model_dir = "my-bert-model"
12
 
13
- config = AutoConfig.from_pretrained(model_dir, num_labels=3, finetuning_task="text-classification")
14
- tokenizer = AutoTokenizer.from_pretrained(model_dir)
15
- model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config)
16
 
17
  def inference(input_text):
18
  inputs = tokenizer.batch_encode_plus(
 
8
  AutoTokenizer,
9
  )
10
 
11
+ model_id = "oliv32/bert-fine-tuned"
12
 
13
+ config = AutoConfig.from_pretrained(model_id, num_labels=3, finetuning_task="text-classification")
14
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
15
+ model = AutoModelForSequenceClassification.from_pretrained(model_id, config=config)
16
 
17
  def inference(input_text):
18
  inputs = tokenizer.batch_encode_plus(