Update app.py

app.py

@@ -28,7 +28,7 @@ def main():
     if st.button("Generate"):
         if not prompt:
             st.warning("Please enter a prompt first.")
-            return  # ✅ Fixed indentation
+            return  # ✅ Make sure this return is correctly indented
 
         st.write("Generating your image...")
         with torch.no_grad():
@@ -40,3 +40,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+