Update app.py

app.py

@@ -48,7 +48,9 @@ class GradioInference():
 
  label = self.classifier(results["text"])[0]["label"]
  wordcloud = WordCloud(width=800, height=400, background_color='white').generate(results["text"])
- return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud
+ wordcloud_image = wordcloud.to_image()
+ 
+ return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud_image
 
  def populate_metadata(self, link):
  self.yt = YouTube(link)
@@ -78,8 +80,9 @@ class GradioInference():
 
  label = self.classifier(results["text"])[0]["label"]
  wordcloud = WordCloud(width=800, height=400, background_color='white').generate(results["text"])
+ wordcloud_image = wordcloud.to_image()
  
- return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud
+ return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud_image
 
 
 gio = GradioInference()
@@ -117,11 +120,11 @@ with block as demo:
  label = gr.Label(label="Sentiment Analysis")
  with gr.Row().style(equal_height=True): 
  # Display the Word Cloud
- wordcloud = gr.Image()
+ wordcloud_image = gr.Image()
  with gr.Row().style(equal_height=True):
  clear = gr.ClearButton([link, title, img, text, summary, keywords, label], scale=1)
  btn = gr.Button("Get video insights", variant='primary', scale=1)
- btn.click(gio, inputs=[link, lang, size], outputs=[text, summary, keywords, label, wordcloud])
+ btn.click(gio, inputs=[link, lang, size], outputs=[text, summary, keywords, label, wordcloud_image])
  link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
 
  with gr.Tab("From Audio file"):