File size: 920 Bytes
748c41a ebd6305 5fd48dd 55b0049 748c41a ebd6305 748c41a ebd6305 748c41a ebd6305 291e47a 748c41a 291e47a 748c41a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import pickle
import numpy as np
import gradio as gr
# install transformers and torch in requirements.txt
from transformers import AutoTokenizer, AutoModelForSequenceClassification
from sklearn.feature_extraction.text import TfidfVectorizer
model = pickle.load(open("model.pkl", "rb"))
vectorizer = pickle.load(open("vectorizer.pkl", "rb"))
def classify_text(inp):
new_question_vector = vectorizer.transform([inp])
prediction = model.predict(new_question_vector)
return str(prediction[0])
iface = gr.Interface(fn=classify_text, inputs="text", outputs="label",title="Tabibu Bot",
interpretation="default", examples=[
["I am feeling depressed"],
["I am feeling anxious"],
["I am feeling stressed"],
["I am feeling sad"],
])
iface.launch() |