storresbusquets commited on
Commit
cad98a9
1 Parent(s): 88c56eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ class GradioInference():
47
  keywords = [x.strip() for x in predicted.split(',') if x.strip()]
48
 
49
  label = self.classifier(results["text"])[0]["label"]
50
- wordcloud = WordCloud(width=800, height=400, background_color='white').generate(results["text"])
51
  wordcloud_image = wordcloud.to_image()
52
 
53
  return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud_image
 
47
  keywords = [x.strip() for x in predicted.split(',') if x.strip()]
48
 
49
  label = self.classifier(results["text"])[0]["label"]
50
+ wordcloud = WordCloud().generate(results["text"])
51
  wordcloud_image = wordcloud.to_image()
52
 
53
  return results["text"], transcription_summary[0]["summary_text"], keywords, label, wordcloud_image