Update app.py

app.py

@@ -73,7 +73,12 @@ def generate_image(prompt):
 
         if isinstance(result, tuple) and len(result) > 0 and isinstance(result[0], str):
             img_str = result[0]
-            img_str += '=' * (-len(img_str) % 4)
+            
+            # Fix padding if necessary
+            missing_padding = len(img_str) % 4
+            if missing_padding:
+                img_str += '=' * (4 - missing_padding)
+                
             img_data = base64.b64decode(img_str)
             image = PILImage.open(BytesIO(img_data))
         elif isinstance(result, tuple) and len(result) > 0 and isinstance(result[0], np.ndarray):