Update app.py

app.py

@@ -13,16 +13,19 @@ def get_caption(image_in):
 
 def get_lcm(prompt):
     client = Client("https://latent-consistency-lcm-lora-for-sdxl.hf.space/")
-    result = client.predict(
-    		prompt,	# str  in 'parameter_5' Textbox component
-    		0.3,	# float (numeric value between 0.0 and 5) in 'Guidance' Slider component
-    		8,	# float (numeric value between 2 and 10) in 'Steps' Slider component
-    		0,	# float (numeric value between 0 and 12013012031030) in 'Seed' Slider component
-    		True,	# bool  in 'Randomize' Checkbox component
-    							api_name="/predict"
+    # Assuming 'client.predict' can return multiple image variations and accepts parameters to specify the number.
+    results = client.predict(
+        prompt,  # str in 'parameter_5' Textbox component
+        0.3,     # float (numeric value between 0.0 and 5) in 'Guidance' Slider component
+        8,       # float (numeric value between 2 and 10) in 'Steps' Slider component
+        0,       # float (numeric value between 0 and 12013012031030) in 'Seed' Slider component
+        True,    # bool in 'Randomize' Checkbox component
+        num_images=4,  # New parameter to request 4 images (this parameter name might be different)
+        api_name="/predict"
     )
-    print(result)
-    return result[0]
+    print(results)
+    return results[:4]  # Return first 4 images in case the API returns more
+
 
 def infer(image_in):
     caption = get_caption(image_in)