File size: 622 Bytes
6ee41b4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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()