Spaces:
Runtime error
Runtime error
File size: 848 Bytes
1beac17 eeb8f8f 1beac17 42afb6c 1beac17 eeb8f8f 1beac17 |
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 pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.cluster import KMeans
labels = {}
def get_sentiment(input_txt, k):
data = input_txt.splitlines()
df = pd.DataFrame(data, columns=["content"])
vec = TfidfVectorizer()
features = vec.fit_transform(df["content"])
kmeans = KMeans(k, n_init=10, random_state=42)
kmeans.fit(features)
df['kmeans'] = kmeans.labels_
labels = df['kmeans'].value_counts(normalize=True).to_dict()
return labels
iface = gr.Interface(fn= get_sentiment,
inputs = [gr.inputs.Textbox(lines=8, placeholder="News Here..."), gr.inputs.Slider(1, 101)],
outputs = ["label"],
title = "Short Document Clustering")
if __name__ == "__main__":
app, local_url, share_url = iface.launch() |