Update app.py

app.py

@@ -25,7 +25,7 @@ pipe = GlmImagePipeline.from_pretrained(
 def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024, height=1024, 
           num_inference_steps=50, guidance_scale=1.5, progress=gr.Progress(track_tqdm=True)):
     """Main inference function"""
-    
+    print("Randomizing seed")
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
@@ -34,7 +34,8 @@ def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024,
     height = (height // 32) * 32
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    
+
+    print("preparing iages")
     # Prepare image list for image-to-image mode
     image_list = None
     if input_images is not None and len(input_images) > 0:
@@ -46,7 +47,7 @@ def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024,
             elif isinstance(img, Image.Image):
                 img = img.convert("RGB")
             image_list.append(img)
-    
+    print("handling kwargs")
     pipe_kwargs = {
         "prompt": prompt,
         "height": height,
@@ -55,11 +56,11 @@ def infer(prompt, input_images=None, seed=42, randomize_seed=False, width=1024,
         "guidance_scale": guidance_scale,
         "generator": generator,
     }
-    
+    print("adding images")
     # Add images for image-to-image mode
     if image_list is not None:
         pipe_kwargs["image"] = image_list
-    
+    print("running kwargs")
     image = pipe(**pipe_kwargs).images[0]
     
     return image, seed
@@ -98,23 +99,6 @@ def update_dimensions_from_image(image_list):
     
     return new_width, new_height
 
-
-# Text-to-Image examples
-examples_t2i = [
-    ["A beautifully designed modern food magazine style dessert recipe illustration, themed around a raspberry mousse cake. Clean and bright layout with a bold black title 'Raspberry Mousse Cake Recipe Guide', soft-lit close-up of the finished cake showing light pink layers with fresh raspberries and mint leaves."],
-    ["A kawaii die-cut sticker of a chubby orange cat, featuring big sparkly eyes and a happy smile with paws raised in greeting and a heart-shaped pink nose. Smooth rounded lines with black outlines and soft gradient shading with pink cheeks."],
-    ["Soaking wet capybara taking shelter under a banana leaf in the rainy jungle, close up photo, water droplets visible on fur, lush green background"],
-    ["A photorealistic portrait of an astronaut in a detailed spacesuit, floating in space with Earth visible in the background, dramatic lighting from the sun"],
-    ["An elegant art deco poster design for a jazz concert, featuring geometric patterns in gold and deep blue, stylized saxophone silhouette, vintage typography"],
-]
-
-# Image-to-Image examples
-examples_i2i = [
-    ["Transform this scene into a cyberpunk cityscape at night with neon lights and flying cars"],
-    ["Replace the background with an underwater coral reef scene with tropical fish"],
-    ["Convert this to a watercolor painting style with soft pastel colors"],
-]
-
 css = """
 #col-container {
     margin: 0 auto;