MrDdz commited on
Commit
7436729
·
1 Parent(s): f24ab5f

app updated

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import torch
7
  @st.cache_resource
8
  def predict_sentiment(text):
9
  # Load the model and tokenizer
10
- model = transformers.AutoModelForSequenceClassification.from_pretrained("MrDdz/mytuned_test_trainer-base-cased1")
11
- tokenizer = transformers.AutoTokenizer.from_pretrained("xlnet-base-cased")
12
 
13
  # Predict the sentiment.
14
  prediction = (text)
 
7
  @st.cache_resource
8
  def predict_sentiment(text):
9
  # Load the model and tokenizer
10
+ model = AutoModelForSequenceClassification.from_pretrained("MrDdz/mytuned_test_trainer-base-cased1")
11
+ tokenizer = AutoTokenizer.from_pretrained("xlnet-base-cased")
12
 
13
  # Predict the sentiment.
14
  prediction = (text)