Update app.py

app.py

@@ -10,6 +10,7 @@ import io
 import csv
 from datetime import datetime
 import os
+from PIL import Image
 
 # Set up translation and summarization models
 translation_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
@@ -78,7 +79,7 @@ def generate_wordcloud(words):
     plt.savefig(buf, format="png")
     plt.close(fig)
     buf.seek(0)
-    return buf
+    return Image.open(buf)
 
 def export_to_csv():
     filename = "feedback_export.csv"