DeeeTeeee01 commited on
Commit
4ded591
·
1 Parent(s): 6874360

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,8 +70,8 @@ import transformers
70
  import torch
71
 
72
  # Load the model and tokenizer
73
- model = transformers.AutoModelForSequenceClassification.from_pretrained("DeeeTeeee01/mytest_trainer_base-cased")
74
- tokenizer = transformers.AutoTokenizer.from_pretrained("DeeeTeeee01/mytest_trainer_base-cased")
75
 
76
  # Define the function for sentiment analysis
77
  @st.cache_resource
 
70
  import torch
71
 
72
  # Load the model and tokenizer
73
+ model = transformers.AutoModelForSequenceClassification.from_pretrained("DeeeTeeee01/mytest_trainer_roberta-base")
74
+ tokenizer = transformers.AutoTokenizer.from_pretrained("DeeeTeeee01/mytest_trainer_roberta-base")
75
 
76
  # Define the function for sentiment analysis
77
  @st.cache_resource