Ehrii commited on
Commit
4ef5369
·
verified ·
1 Parent(s): 51331f6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -16,10 +16,12 @@ app = FastAPI()
16
  # Load the original tokenizer from the base model
17
  original_tokenizer = AutoTokenizer.from_pretrained("tabularisai/multilingual-sentiment-analysis")
18
 
 
 
19
  # Load the fine-tuned model and pass the tokenizer explicitly
20
  multilingual_model = pipeline(
21
  "sentiment-analysis",
22
- model="Ehrii/sentiment",
23
  tokenizer=original_tokenizer
24
  )
25
 
 
16
  # Load the original tokenizer from the base model
17
  original_tokenizer = AutoTokenizer.from_pretrained("tabularisai/multilingual-sentiment-analysis")
18
 
19
+ model_name = "Ehrii/multimodel" # Replace with your username/repo name
20
+
21
  # Load the fine-tuned model and pass the tokenizer explicitly
22
  multilingual_model = pipeline(
23
  "sentiment-analysis",
24
+ model=model_name,
25
  tokenizer=original_tokenizer
26
  )
27