Vineedhar commited on
Commit
e5b0a02
·
verified ·
1 Parent(s): a66b403

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -3,10 +3,13 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipe
3
  import tensorflow as tf
4
  tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
5
 
 
 
 
 
 
6
  # Load pre-trained model and tokenizer
7
- saved_directory = "models/cardiffnlp/twitter-xlm-roberta-base-sentiment"
8
- tokenizer = AutoTokenizer.from_pretrained(saved_directory)
9
- model = AutoModelForSequenceClassification.from_pretrained(saved_directory)
10
  nlp = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
11
 
12
  # Define function to analyze sentiment
 
3
  import tensorflow as tf
4
  tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
5
 
6
+ # Load model directly
7
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
8
+
9
+ tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
10
+ model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
11
  # Load pre-trained model and tokenizer
12
+
 
 
13
  nlp = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
14
 
15
  # Define function to analyze sentiment