Hugging Face's logo

Spaces:
waleed-12
/
Sentiment-Analyzer
Sleeping

App Files Community
waleed-12 commited on
Commit
7b28085
·
verified ·
1 Parent(s): 35f7c21

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import torch.nn.functional as F
5
 
6
  # Load model and tokenizer from Hugging Face
7
- MODEL_NAME = "imrgurmeet/fine-tuned-sentiment-model"
8
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
9
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
10
 
 
4
  import torch.nn.functional as F
5
 
6
  # Load model and tokenizer from Hugging Face
7
+ MODEL_NAME = "nlptown/bert-base-multilingual-uncased-sentiment"
8
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
9
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
10