Update app.py

app.py

@@ -15,6 +15,7 @@ import random
 from PIL import Image
 import gradio as gr
 openai.api_key ='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'
+from io import BytesIO
 
 # def split_article(article_text):
 #     words = article_text.split()    
@@ -393,21 +394,25 @@ openai.api_key ='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'
 #                                 "Imagine a colossal arena where AI-powered chatbots engage in an epic battle for dominance. The arena is a futuristic coliseum with digital screens lining the walls, broadcasting the ongoing clashes between chatbot warriors. Each chatbot represents a different AI assistant, armed with unique capabilities and strategies. The crowd roars with anticipation as they place their bets on which chatbot will emerge victorious. Describe the electrifying atmosphere of competition and the spectacle of Chatbot Wars.",
 #                                 "Picture a digital revolution square, where AI chatbots lead the charge for a new era of digital communication. The square is a vibrant hub of innovation, with holographic screens displaying the transformation of messaging through AI. AI chatbot leaders, with visionary personas, address the gathered crowd, sharing their insights and predictions for the future of digital communication. Describe the electric atmosphere of change and the role AI chatbots play in reshaping the digital landscape."])
 
+
+
 def get_image(prompt): 
-    api_token="r8_DwF25VNEJXEdjULfkV5gBjYnnJ6pwlg4OmDg7"
+    api_token = "r8_DwF25VNEJXEdjULfkV5gBjYnnJ6pwlg4OmDg7"
     client = replicate.Client(api_token=api_token)
     output = client.run(
         "stability-ai/sdxl:a00d0b7dcbb9c3fbb34ba87d2d5b46c56969c84a628bf778a7fdaec30b1b99c5",
-        input={"prompt":'Phantasmal iridescent, vibrant color, high contrast, award winning, trending in artstation, digital art,' + prompt.strip(),
-               "negative_prompt":'NSFW, cityscape',
-              "width": 1648,
-              "height":1024}
+        input={
+            "prompt": 'Phantasmal iridescent, vibrant color, high contrast, award winning, trending in artstation, digital art,' + prompt.strip(),
+            "negative_prompt": 'NSFW, cityscape',
+            "width": 1648,
+            "height": 1024
+        }
     )
     gambar = output
     image_url = gambar[0]
     response = requests.get(image_url)
-    return response.content
-
+    img = Image.open(BytesIO(response.content))
+    return img
         
 #     return tags, content, url #image, image_data