jbraha commited on
Commit
3cf166b
1 Parent(s): 8e37781
Files changed (2) hide show
  1. .gitignore +0 -7
  2. app.py +1 -1
.gitignore CHANGED
@@ -1,9 +1,2 @@
1
  results/**
2
- <<<<<<< HEAD
3
- <<<<<<< HEAD
4
  data/**
5
- =======
6
- >>>>>>> f375d50 ('mint autosave')
7
- =======
8
- data/**
9
- >>>>>>> 56fcd6b ('mint autosave')
 
1
  results/**
 
 
2
  data/**
 
 
 
 
 
app.py CHANGED
@@ -28,7 +28,7 @@ option = st.selectbox(
28
 
29
  if option == 'Fine-Tuned':
30
  model = AutoModelForSequenceClassification.from_pretrained(fine_tuned)
31
- tokenizer = AutoTokenizer.from_pretrained(fine_tuned)
32
  classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
33
  elif option == 'Roberta':
34
  model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
 
28
 
29
  if option == 'Fine-Tuned':
30
  model = AutoModelForSequenceClassification.from_pretrained(fine_tuned)
31
+ tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
32
  classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
33
  elif option == 'Roberta':
34
  model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")