Update app.py

app.py

@@ -81,16 +81,16 @@ st.caption('An app to generate images based on text prompts with a :blue[_Stable
 prompt = st.text_input('Input the prompt desired')
 
 if len(prompt) > 0:
-  st.markdown(f"""
-  This will show an image using **stable diffusion** of the desired {prompt} entered:
-  """)
-  print(prompt)
-  # Create a spinner to show the image is being generated
-  with st.spinner('Generating image based on prompt'):
-    sd = StableDiffusionLoader(prompt)
-    sd.generate_image_from_prompt(save_location=SAVE_LOCATION)
-    st.success('Generated stable diffusion model')
-
-# Open and display the image on the site
-image = Image.open(SAVE_LOCATION)
-st.image(image)  
+    st.markdown(f"""
+    This will show an image using **stable diffusion** of the desired {prompt} entered:
+    """)
+    print(prompt)
+    # Create a spinner to show the image is being generated
+    with st.spinner('Generating image based on prompt'):
+        sd = StableDiffusionLoader(prompt)
+        sd.generate_image_from_prompt(save_location=SAVE_LOCATION)
+        st.success('Generated stable diffusion model')
+        
+        # Open and display the image on the site
+        image = Image.open(SAVE_LOCATION)
+        st.image(image)