Update app.py

app.py

@@ -535,23 +535,24 @@ class UltraSupremeOptimizer:
         except Exception as e:
             logger.error(f"Initialization error: {e}")
             return False
-            def optimize_image(self, image):
-        if image is None:
-            return None
             
-        if isinstance(image, np.ndarray):
-            image = Image.fromarray(image)
-        elif not isinstance(image, Image.Image):
-            image = Image.open(image)
-        
-        if image.mode != 'RGB':
-            image = image.convert('RGB')
-        
-        max_size = 768 if self.device != "cpu" else 512
-        if image.size[0] > max_size or image.size[1] > max_size:
-            image.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
-        
-        return image
+def optimize_image(self, image):
+    if image is None:
+        return None
+        
+    if isinstance(image, np.ndarray):
+        image = Image.fromarray(image)
+    elif not isinstance(image, Image.Image):
+        image = Image.open(image)
+    
+    if image.mode != 'RGB':
+        image = image.convert('RGB')
+    
+    max_size = 768 if self.device != "cpu" else 512
+    if image.size[0] > max_size or image.size[1] > max_size:
+        image.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
+    
+    return image
     
     @spaces.GPU
     def generate_ultra_supreme_prompt(self, image):