WordCLoud / app.py
therayz1's picture
Update app.py
5c07633 verified
raw
history blame contribute delete
798 Bytes
# Step 2: Import libraries
import gradio as gr
from wordcloud import WordCloud
import matplotlib.pyplot as plt
# Step 3: Define the word cloud generator function
def generate_word_cloud(text):
wordcloud = WordCloud(width=800, height=400, background_color='white').generate(text)
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')
plt.savefig("wordcloud.png")
return "wordcloud.png"
# Step 4: Set up the Gradio interface
iface = gr.Interface(
fn=generate_word_cloud,
inputs=gr.components.Textbox(lines=10, placeholder="Enter text here..."),
outputs="image",
title="Word Cloud Generator",
description="Generate a word cloud from the input text"
)
# Step 5: Launch the interface in Colab
iface.launch(share=True)