JohnDoee commited on
Commit
2a34452
·
1 Parent(s): 76966a6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -16,12 +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
- 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
 
 
16
 
17
  # Load the original tokenizer from the base model
18
  original_tokenizer = AutoTokenizer.from_pretrained("tabularisai/multilingual-sentiment-analysis")
19
+ mode_name = "johndoee/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_name),
25
  tokenizer=original_tokenizer
26
  )
27