Update app.py

app.py

@@ -30,7 +30,7 @@ def run_dif(prompt,im_path,model_drop,cnt):
     url = base_url+im_path
     init_image = load_image(url)
     
-    prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+    #prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
     
     # pass prompt and image to pipeline
     #image = pipeline(prompt, image=init_image, strength=0.8,guidance_scale=8.0,negative_prompt=negative_prompt,num_inference_steps=50).images[0]
@@ -130,18 +130,6 @@ def thread_dif(out_prompt,mod):
         return out_box,html_out
 
 
-            
-def start_threads(prompt):
-    t1 = threading.Thread(target=thread_dif, args=(prompt,0))
-    t2 = threading.Thread(target=thread_dif, args=(prompt,1))
-    t1.start()
-    t2.start()
-    print (t1)
-    print (t2)
-    a1,a2=t1.result()
-    b1,b2=t2.result()
-    return a1,a2
-            
 css="""
 .grid_class{
 display:flex;
@@ -152,16 +140,43 @@ min-width:200px;
 }
 
 """
+
+def load_im(img):
+    im_box=[]
+    im = Image.open(img)
+    width, height = im.size
+    new_w=int(width/10)
+    new_h=int(height/10)
+
+
+    
+    # Setting the points for cropped image
+    left = 0
+    top = 0
+    right = new_w
+    bottom = new_h
+     
+    # Cropped image of above dimension
+    # (It will not change original image)
+    im1 = im.crop((left, top, right, bottom))
+    im_box.append(im1)
+    return im_box
+
+
 with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column():
             inp=gr.Textbox(label="Prompt")
             btn=gr.Button()
-        inp_im=gr.Image(type='filepath')
+        with gr.Column():
+            inp_im=gr.Image(type='filepath')
+            im_btn=gr.Button()
     with gr.Row():
         model_drop=gr.Dropdown(label="Models", choices=models, type='index', value=models[0])
         cnt = gr.Number(value=1)
     out_html=gr.HTML()
     outp=gr.Gallery()
+
+    im_btn.click(load_im,inp_im,outp)
     btn.click(run_dif,[inp,inp_im,model_drop,cnt],[outp,out_html])
 app.launch()