Spaces:
Build error
Build error
File size: 668 Bytes
1f4d723 71669c4 1f4d723 |
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 29 30 31 32 |
import gradio as gr
import pickle
from nltk.tokenize import word_tokenize
import nltk
nltk.download('all')
with open('LogisticRegression_classifier.pickle','rb') as fp:
LogisticRegression_classifier = pickle.load(fp)
with open('word_features5k.pickle','rb') as fp:
word_features = pickle.load(fp)
def find_features(news):
words = word_tokenize(news)
features = {}
for w in word_features:
features[w] = (w in words)
return features
def fn(news):
return LogisticRegression_classifier.classify(find_features(news))
iface = gr.Interface(
fn = fn,
inputs = 'text',
outputs = 'text'
)
url = iface.launch()
# return url[] |