Update app.py

app.py

@@ -394,32 +394,28 @@ openai.api_key ='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'
 #                                 "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): 
-    try:
-        os.environ["REPLICATE_API_TOKEN"] = "r8_KUXOnCBFpGtsxpL2MNUMi9CVYulQjEc0pOlhs"
-        sdxl_model = replicate.models.get("stability-ai/sdxl")
-        sdxl_version = sdxl_model.versions.get("a00d0b7dcbb9c3fbb34ba87d2d5b46c56969c84a628bf778a7fdaec30b1b99c5")
-    
-        prediction = replicate.predictions.create(version=sdxl_version,
-            input={"prompt":'Phantasmal iridescent, vibrant color, high contrast, award winning, trending in artstation, digital art' + prompt,
-                    "negative_prompt":'NSFW, cityscape',
-                    "width": 1648,
-                    "height":1024}
-            )  
-        prediction
-        prediction.reload()
-        prediction.wait()
-    
-        if prediction.status == 'failed':
-            print(f"Error: {prediction.error}")
-            print(f"Last Log")
-            print(prediction.logs)
-        elif ((prediction.status == 'succeeded') | (prediction.output != None)):
-            response = requests.get(prediction.output[0])
-            image_data = base64.b64encode(response.content)
-            return image_data
-    except:
-        image_data = None
-        return image_data
+    os.environ["REPLICATE_API_TOKEN"] = "r8_KUXOnCBFpGtsxpL2MNUMi9CVYulQjEc0pOlhs"
+    sdxl_model = replicate.models.get("stability-ai/sdxl")
+    sdxl_version = sdxl_model.versions.get("a00d0b7dcbb9c3fbb34ba87d2d5b46c56969c84a628bf778a7fdaec30b1b99c5")
+
+    prediction = replicate.predictions.create(version=sdxl_version,
+        input={"prompt":'Phantasmal iridescent, vibrant color, high contrast, award winning, trending in artstation, digital art' + prompt,
+                "negative_prompt":'NSFW, cityscape',
+                "width": 1648,
+                "height":1024}
+        )  
+    prediction
+    prediction.reload()
+    prediction.wait()
+
+    if prediction.status == 'failed':
+        print(f"Error: {prediction.error}")
+        print(f"Last Log")
+        print(prediction.logs)
+    elif ((prediction.status == 'succeeded') | (prediction.output != None)):
+        response = requests.get(prediction.output[0])
+        return response.content
+
         
 #     return tags, content, url #image, image_data