Morris commited on
Commit
36c82f7
·
1 Parent(s): 009c4dd

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  import pandas as pd
3
  from PIL import Image
4
- from transformers import pipeline
5
  pipe = pipeline('text-classification', model='tiedaar/short-answer-classification')
6
  bleurt_pipe = pipeline('text-classification', model="vaiibhavgupta/finetuned-bleurt-large")
7
  tokenizer = AutoTokenizer.from_pretrained("vaiibhavgupta/finetuned-bleurt-large")
 
1
  import streamlit as st
2
  import pandas as pd
3
  from PIL import Image
4
+ from transformers import pipeline, AutoTokenizer
5
  pipe = pipeline('text-classification', model='tiedaar/short-answer-classification')
6
  bleurt_pipe = pipeline('text-classification', model="vaiibhavgupta/finetuned-bleurt-large")
7
  tokenizer = AutoTokenizer.from_pretrained("vaiibhavgupta/finetuned-bleurt-large")