Orawan commited on
Commit
5e5b20f
·
1 Parent(s): a6a4045

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,12 +1,11 @@
1
  import streamlit as st
2
- # Load model directly
3
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
 
5
  tokenizer = AutoTokenizer.from_pretrained("OatNapat/finetuned_yelp")
6
  model = AutoModelForSequenceClassification.from_pretrained("OatNapat/finetuned_yelp")
7
- # Use a pipeline as a high-level helper
8
- from transformers import pipeline
9
 
10
- nlp = pipeline("text-classification", model="zfox/finetuning-sentiment-model-3000-samples")
 
11
 
12
  text = st.text_input('กรุณาถอดรองเท้า')
 
1
  import streamlit as st
2
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
3
 
4
+ # Load the tokenizer and model
5
  tokenizer = AutoTokenizer.from_pretrained("OatNapat/finetuned_yelp")
6
  model = AutoModelForSequenceClassification.from_pretrained("OatNapat/finetuned_yelp")
 
 
7
 
8
+ # Create a sentiment analysis pipeline with the explicit tokenizer
9
+ nlp = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
10
 
11
  text = st.text_input('กรุณาถอดรองเท้า')