storresbusquets commited on
Commit
9e1df22
·
1 Parent(s): d57300e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -118,7 +118,7 @@ class GradioInference:
118
  label = self.classifier(summary)[0]["label"]
119
 
120
  # Generate WordCloud object
121
- wordcloud = WordCloud().generate(results["text"], colormap = "Oranges")
122
 
123
  # WordCloud image to display
124
  wordcloud_image = wordcloud.to_image()
@@ -223,9 +223,8 @@ class GradioInference:
223
  label = self.classifier(summary)[0]["label"]
224
 
225
  # WordCloud object
226
- wordcloud = WordCloud().generate(
227
- results["text"],
228
- colormap = "Oranges"
229
  )
230
  wordcloud_image = wordcloud.to_image()
231
 
 
118
  label = self.classifier(summary)[0]["label"]
119
 
120
  # Generate WordCloud object
121
+ wordcloud = WordCloud(colormap = "Oranges").generate(results["text"])
122
 
123
  # WordCloud image to display
124
  wordcloud_image = wordcloud.to_image()
 
223
  label = self.classifier(summary)[0]["label"]
224
 
225
  # WordCloud object
226
+ wordcloud = WordCloud(colormap = "Oranges").generate(
227
+ results["text"]
 
228
  )
229
  wordcloud_image = wordcloud.to_image()
230