Update app.py

app.py

@@ -33,6 +33,7 @@ controlnet_lineart = ControlNetModel.from_pretrained(
     "fffiloni/cn_malgras_test_11",
     torch_dtype=torch.float16,
     variant="fp16",
+    token=hf_token
 )
 
 def check_use_custom_or_no(value):
@@ -204,7 +205,8 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
 
     images[0].save(f"result.png")
 
-    return f"result.png", seed
+    #return f"result.png", seed
+    return [image] + images[0], seed
 
 css="""
 #col-container{
@@ -347,7 +349,7 @@ with gr.Blocks(css=css) as demo:
         
         submit_btn = gr.Button("Submit")
         
-        result = gr.Image(label="Result")
+        result = gr.Gallery(label="Result")
     
     use_custom_model.change(
         fn = check_use_custom_or_no,