RICHARDMENSAH commited on
Commit
5832a58
·
1 Parent(s): 468a2c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -3,8 +3,13 @@ import transformers
3
  import torch
4
 
5
  # Load the model and tokenizer
6
- model = transformers.AutoModelForSequenceClassification.from_pretrained("RICHARDMENSAH/twitter_xlm_roberta_base")
7
- tokenizer = transformers.AutoTokenizer.from_pretrained("RICHARDMENSAH/twitter_xlm_roberta_base")
 
 
 
 
 
8
 
9
  # Define the function for sentiment analysis
10
  @st.cache_resource
 
3
  import torch
4
 
5
  # Load the model and tokenizer
6
+ #model = transformers.AutoModelForSequenceClassification.from_pretrained("RICHARDMENSAH/twitter_xlm_roberta_base")
7
+ # tokenizer = transformers.AutoTokenizer.from_pretrained("RICHARDMENSAH/twitter_xlm_roberta_base")
8
+
9
+ model_name = "RICHARDMENSAH/twitter_xlm_roberta_base"
10
+ tokenizer_name = "RICHARDMENSAH/twitter_xlm_roberta_base"
11
+
12
+ model = transformers.pipeline("text-classification", model=model_name, tokenizer=tokenizer_name)
13
 
14
  # Define the function for sentiment analysis
15
  @st.cache_resource