Update app.py

app.py

@@ -6,7 +6,6 @@ from PIL import Image
 import gradio as gr
 from pathlib import Path 
 from zipfile import ZipFile
-import dif
 from huggingface_hub import (create_repo,get_full_repo_name,upload_file,CommitOperationAdd,HfApi)
 uid = uuid.uuid4()
 
@@ -455,7 +454,7 @@ def gen_im(prompt,model_choice=5,count=1):
     output_list=[]
     while t<int(count):
         proc1=models2[int(model_choice)]
-        output1=proc1(inputs)
+        output1=proc1(prompt)
         output_list.append(output1)
         inputs=f"{inputs} "
         t+=1
@@ -510,7 +509,7 @@ with gr.Blocks() as build_app:
                     with gr.Row():
                         magic_btn=gr.Button("Magic Prompt")
                         gen_btn=gr.Button("Generate Images")
-                    gen_im_out=gr.Gallery()
+                    gen_im_out=gr.Gallery(columns=2)
                 c_image=gr.Image(label="Cropped Image", type="filepath",interactive=False)
                 c_slider=gr.Slider(0,220,value=0,step=1,label="Adjust Vertical Crop")
                 c_upbtn=gr.Button("Update Crop")