Update app.py

app.py

@@ -17,17 +17,14 @@ def generate_image(prompt, seed=0, negative_prompt="", model="Deliberate", sampl
         "steps": steps
     }
     response = requests.post(api_url, data=data)
-    if response.status_code == 200:
-        # Decode the base64 image string
-        img_bytes = base64.b64decode(response.json()["url"])
-        
-        # Open the image using Pillow
-        img = Image.open(io.BytesIO(img_bytes))
-        
-        # Return the image object
-        return img
-    else:
-        return None
+    # Decode the base64 image string
+    img_bytes = base64.b64decode(response.json()["url"])
+    
+    # Open the image using Pillow
+    img = Image.open(io.BytesIO(img_bytes))
+    
+    # Return the image object
+    return img
 
 inputs = [
     gr.inputs.Textbox(label="Prompt"),