Update app.py

app.py

@@ -38,6 +38,7 @@ def infer(prompt, api_key):
                                                  # Defaults to k_dpmpp_2m if not specified. Clip Guidance only supports ancestral samplers.
                                                  # (Available Samplers: ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_dpmpp_2s_ancestral, k_lms, k_dpmpp_2m)
 )
+    results = []
     for resp in answers:
         for artifact in resp.artifacts:
             if artifact.finish_reason == generation.FILTER:
@@ -46,8 +47,9 @@ def infer(prompt, api_key):
                 "Please modify the prompt and try again.")
             if artifact.type == generation.ARTIFACT_IMAGE:
                 img = Image.open(io.BytesIO(artifact.binary))
+                results.append(img)
     
-    return [img]
+    return results
 
 css = """
         .gradio-container {