Update app.py

app.py

@@ -22,6 +22,7 @@ def query(payload, model):
 
 def hf_inference(prompt, negative, model, steps, sampler, guidance, width, height, seed):
     try:
+        images=[]
         image_bytes = query(payload={
             "inputs": f"{prompt}",
             "parameters": {
@@ -33,7 +34,8 @@ def hf_inference(prompt, negative, model, steps, sampler, guidance, width, heigh
             },
         }, model=model)
         image = Image.open(io.BytesIO(image_bytes))
-        return image
+        images.append(image)
+        return images
     except PIL.UnidentifiedImageError:
         gr.Warning("This model is not loaded now. Try others models.")
 
@@ -368,7 +370,8 @@ with gr.Blocks(css=css) as demo:
                                       concurrency_limit=64)
         with gr.Tab("HuggingFace Inference"):
             with gr.Row():
-                hf_model = gr.Dropdown(label="HuggingFace checkpoint", choices=["runwayml/stable-diffusion-v1-5", "stabilityai/stable-diffusion-2-1", "dataautogpt3/OpenDalleV1.1", "CompVis/stable-diffusion-v1-4", "playgroundai/playground-v2-1024px-aesthetic", "prompthero/openjourney", "openskyml/dreamdrop-v1", "SG161222/Realistic_Vision_V1.4", "digiplay/AbsoluteReality_v1.8.1", "openskyml/dalle-3-xl", "Lykon/dreamshaper-7", "Pclanglais/Mickey-1928"], value="runwayml/stable-diffusion-v1-5", allow_custom_value=False, interactive=True)
+                gr.Markdown("Add your model from HF.co, enter model ID.")
+                hf_model = gr.Dropdown(label="HuggingFace checkpoint", choices=["runwayml/stable-diffusion-v1-5", "stabilityai/stable-diffusion-2-1", "dataautogpt3/OpenDalleV1.1", "CompVis/stable-diffusion-v1-4", "playgroundai/playground-v2-1024px-aesthetic", "prompthero/openjourney", "openskyml/dreamdrop-v1", "SG161222/Realistic_Vision_V1.4", "digiplay/AbsoluteReality_v1.8.1", "openskyml/dalle-3-xl", "Lykon/dreamshaper-7", "Pclanglais/Mickey-1928"], value="runwayml/stable-diffusion-v1-5", allow_custom_value=True, interactive=True)
             with gr.Row():
                 with gr.Column(scale=6, min_width=600):
                     hf_prompt = gr.Textbox("space warrior, beautiful, female, ultrarealistic, soft lighting, 8k",
@@ -399,7 +402,7 @@ with gr.Blocks(css=css) as demo:
                         hf_seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    hf_image_output = gr.Image(show_label=False, type="pil", interactive=False)
+                    hf_image_output = gr.Gallery(show_label=False, preview=True, columns=4, allow_preview=True)
                     #hf_send_to_img2img = gr.Button(value="Send to img2img")
 
             hf_text_button.click(hf_inference, inputs=[hf_prompt, hf_negative_prompt, hf_model, hf_steps, sampler, hf_cfg_scale, hf_width, hf_height,