diff --git a/apps/flux/app.py b/apps/flux/app.py
index 74dee4d3..7027843f 100644
--- a/apps/flux/app.py
+++ b/apps/flux/app.py
@@ -1,20 +1,19 @@
-import os
-import base64
 import shutil
-import requests
 import replicate
 
 from PIL import Image
 from io import BytesIO
 
-
 def flux_image(prompt, filename):
-  output = replicate.run("black-forest-labs/flux-dev", input={"prompt": prompt})
+  try:
+    output = replicate.run("black-forest-labs/flux-dev", input={"prompt": prompt})
+  except Exception as e:
+    print("Couldn't generate that image. Please try a different prompt.")
+    print(f"Error: {e}")
 
-  header, encoded = output[0].url.split(',', 1)
+  encoded_image = output[0].read()
+  image = Image.open(BytesIO(encoded_image))
 
-  data = base64.b64decode(encoded)
-  image = Image.open(BytesIO(data))
   image.save(filename, format="JPEG")
 
   shutil.copy(filename, f".storage/{filename}")
diff --git a/apps/flux/requirements.txt b/apps/flux/requirements.txt
index 4dd24935..80985dfd 100644
--- a/apps/flux/requirements.txt
+++ b/apps/flux/requirements.txt
@@ -1 +1 @@
-replicate
\ No newline at end of file
+replicate==1.0.0
\ No newline at end of file