JohnDoee commited on
Commit
2d3bcab
·
1 Parent(s): 9232095

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -16,11 +16,12 @@ app = FastAPI()
16
 
17
  # Load the original tokenizer from the base model
18
  original_tokenizer = AutoTokenizer.from_pretrained("tabularisai/multilingual-sentiment-analysis")
 
19
 
20
  # Load the fine-tuned model (johndoee/sentiment) and pass the tokenizer explicitly
21
  multilingual_model = pipeline(
22
  "sentiment-analysis",
23
- model="./sentiment",
24
  tokenizer=original_tokenizer
25
  )
26
 
 
16
 
17
  # Load the original tokenizer from the base model
18
  original_tokenizer = AutoTokenizer.from_pretrained("tabularisai/multilingual-sentiment-analysis")
19
+ model_path = "sentiment" # or "./sentiment"
20
 
21
  # Load the fine-tuned model (johndoee/sentiment) and pass the tokenizer explicitly
22
  multilingual_model = pipeline(
23
  "sentiment-analysis",
24
+ model= AutoModelForSequenceClassification.from_pretrained(model_path),
25
  tokenizer=original_tokenizer
26
  )
27