Update README.

README.md

@@ -9,14 +9,10 @@ app_file: app.py
 pinned: false
 ---
 
+<img src="https://www.infinitescript.com/projects/CityDreamer/CityDreamer-Logo.png" style="height: 150px; margin: 0 auto">
+
 Official demo for **[CityDreamer: Compositional Generative Model of Unbounded 3D Cities](https://github.com/hzxie/city-dreamer) (CVPR 2024).**
 
 - 🔥 CityDreamer is a unbounded 3D city generator.
 - 🤗 Try CityDreamer to generate photolistic 3D cities.
 - ⚠️ Due to the limited computational resources at Hugging Face, this demo only generates **A SINGLE IMAGE** based on the New York City layout. If you wish to experience more comprehensive functionality, please visit the demo on [GitHub](https://github.com/hzxie/city-dreamer?tab=readme-ov-file#iterative-demo-%EF%B8%8F).
-
-<p style="color:red">
-❕IMPORTANT NOTE: The HF team and we are currently working on migrating to ZeroGPU, so the demo may be temporarily unavailable.<br>
-Track the debugging progress on <a href="https://huggingface.co/spaces/zero-gpu-explorers/README/discussions/111">the community of zero-gpu-explorers</a>.
-</p>
-

app.py

@@ -4,7 +4,7 @@
 # @Author: Haozhe Xie
 # @Date:   2024-03-02 16:30:00
 # @Last Modified by: Haozhe Xie
-# @Last Modified at: 2024-09-22 21:11:51
+# @Last Modified at: 2024-09-25 09:09:53
 # @Email:  root@haozhexie.com
 
 import gradio as gr
@@ -36,6 +36,7 @@ import torch
 if os.environ.get("SPACES_ZERO_GPU") is not None:
     import spaces
 else:
+
     class spaces:
         @staticmethod
         def GPU(func):
@@ -45,6 +46,7 @@ else:
 
             return wrapper
 
+
 # Fix: ssl.SSLCertVerificationError: [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed
 ssl._create_default_https_context = ssl._create_unverified_context
 # Import CityDreamer modules
@@ -151,7 +153,7 @@ def get_generated_city(
 
 
 def main(debug):
-    title = "CityDreamer Demo 🏙️"
+    title = "Compositional Generative Model of Unbounded 3D Cities"
     with open("README.md", "r") as f:
         markdown = f.read()
         desc = markdown[markdown.rfind("---") + 3 :]