Update app.py

app.py

@@ -47,8 +47,8 @@ class GradioInference():
  keywords = [x.strip() for x in predicted.split(',') if x.strip()]
 
  label = self.classifier(results["text"])[0]["label"]
- 
- return results["text"], transcription_summary[0]["summary_text"], keywords, label
+ wordcloud = WordCloud(width=800, height=400, background_color='white').generate(results["text"])
+ return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud
 
  def populate_metadata(self, link):
  self.yt = YouTube(link)
@@ -139,7 +139,7 @@ with block as demo:
  with gr.Row().style(equal_height=True):
  clear = gr.ClearButton([text], scale=1)
  btn = gr.Button("Get video insights", variant='primary', scale=1) # Updated button label
- btn.click(gio.from_audio_input, inputs=[lang, size, audio_file], outputs=[text, summary, keywords, label])
+ btn.click(gio.from_audio_input, inputs=[lang, size, audio_file], outputs=[text, summary, keywords, label, wordcloud])
  
 
 with block: