sid22669's picture
Upload 5 files
6ee41b4 verified
raw
history blame contribute delete
622 Bytes
import gradio as gr
import joblib
with open('intent_classifier', 'rb') as file:
model = joblib.load(file)
with open('c_intent_vectorizer', 'rb') as file:
vectorizer = joblib.load(file)
def preprocess_input(data):
tokens = data.lower().split()
filtered = [text for text in tokens if text.isalpha()]
return ' '.join(filtered)
def predict(text):
model_input = preprocess_input(text)
vectorized_input = vectorizer.transform([model_input])
prediction = model.predict(vectorized_input)
return prediction
demo = gr.Interface(fn=predict, inputs="textbox", outputs="textbox")
demo.launch()