Update app.py

app.py

@@ -118,10 +118,10 @@ class GradioInference:
         label = self.classifier(summary)[0]["label"]
         
         # Generate WordCloud object
-        wordcloud = WordCloud().generate(results["text"])
+        wordcloud = WordCloud().generate(results["text"], colormap = "Oranges")
 
         # WordCloud image to display
-        wordcloud_image = wordcloud.to_image(colormap = "Oranges")
+        wordcloud_image = wordcloud.to_image()
 
         if lang == "english":
             return (
@@ -224,9 +224,10 @@ class GradioInference:
         
         # WordCloud object
         wordcloud = WordCloud().generate(
-            results["text"]
+            results["text"],
+            colormap = "Oranges"
         )
-        wordcloud_image = wordcloud.to_image(colormap = "Oranges")
+        wordcloud_image = wordcloud.to_image()
 
         if lang == "english":
             return (