Update app.py

app.py

@@ -69,18 +69,19 @@ elif selected_choice=="Captured":
             image_file = Image.open(uploaded_file)
             image_file = image_file.resize((360,360), Image.LANCZOS)
             st.image(image_file)
+            
+            prompt = st.text_input("write prompt for depth to image")
 
-            if st.button("Get Variations"):
-                        
-                depth2image = StableDiffusionDepth2ImgPipeline.from_pretrained(
+            depth2image = StableDiffusionDepth2ImgPipeline.from_pretrained(
                 "stabilityai/stable-diffusion-2-depth",
                 torch_dtype=torch.float32,
                 ).to("cpu")
 
-                prompt = st.text_input("write prompt for depth to image")
                 
-                image = depth2image(prompt=prompt, image=image_file,
+            image = depth2image(prompt=prompt, image=image_file,
                                     strength=0.8,num_inference_steps=10).images[0]
-                
+            
+            if st.button("Get Variations"):
+                        
                 with col2:
                     st.image(image)