Suratsada commited on
Commit
96e988e
1 Parent(s): 567a3ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,9 +1,9 @@
1
- !python -m pip install transformers --quiet
2
- from transformers import pipeline
 
3
 
4
- classifier = pipeline("sentiment-analysis",
5
- model="distilbert-base-uncased-finetuned-sst-2-english")
6
-
7
  def main():
8
  st.title("classifier")
9
 
@@ -11,7 +11,7 @@ def main():
11
  X = st.text_input("")
12
  clicked = st.form_submit_button("Predict")
13
  if clicked:
14
- result=classifier(X)
15
  st.success(result)
16
 
17
  if __name__ == "__main__":
 
1
+ import joblib
2
+ import pandas as pd
3
+ import streamlit as st
4
 
5
+ model = joblib.load('model.joblib')
6
+
 
7
  def main():
8
  st.title("classifier")
9
 
 
11
  X = st.text_input("")
12
  clicked = st.form_submit_button("Predict")
13
  if clicked:
14
+ result=model(X)
15
  st.success(result)
16
 
17
  if __name__ == "__main__":