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: