Donlapark commited on
Commit
19de662
·
1 Parent(s): 066162b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import streamlit as st
2
- from transformers import pipeline
 
3
 
4
- classifier = pipeline("sentiment-analysis", model="Donlapark/finetuned_yelp")
 
 
5
  def main():
6
  st.title("Yelp review")
7
 
@@ -10,8 +13,9 @@ def main():
10
  # clicked==True only when the button is clicked
11
  clicked = st.form_submit_button("Submit")
12
  if clicked:
13
- results = classifier([text])
14
- st.json(results)
 
15
 
16
  if __name__ == "__main__":
17
  main()
 
1
  import streamlit as st
2
+ import torch
3
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
5
+ saved_model = "Donlapark/finetuned_yelp"
6
+ tokenizer = AutoTokenizer.from_pretrained("Donlapark/finetuned_yelp")
7
+ model = AutoModelForSequenceClassification.from_pretrained("Donlapark/finetuned_yelp")
8
  def main():
9
  st.title("Yelp review")
10
 
 
13
  # clicked==True only when the button is clicked
14
  clicked = st.form_submit_button("Submit")
15
  if clicked:
16
+ sentence = tokenizer(text, return_tensors="pt")
17
+ results = torch.softmax(model(**sentence).logits, axis=1).numpy()[0]
18
+ st.write(f"Predicted review: {results.argmax()}, Score: {results.max()}")
19
 
20
  if __name__ == "__main__":
21
  main()