Spaces:
Running
on
A10G
Running
on
A10G
# -*- coding: utf-8 -*- | |
# | |
# @File: app.py | |
# @Author: Haozhe Xie | |
# @Date: 2024-03-02 16:30:00 | |
# @Last Modified by: Haozhe Xie | |
# @Last Modified at: 2024-03-02 22:33:17 | |
# @Email: root@haozhexie.com | |
import streamlit as st | |
from PIL import Image | |
LOGGER = st.logger.get_logger(__name__) | |
def setup_runtime_env(): | |
pass | |
def get_models(): | |
return None, None | |
def get_generated_city(radius, altitude, azimuth): | |
pass | |
def main(fgm, bgm): | |
st.set_page_config( | |
page_title="CityDreamer Demo", | |
page_icon="๐๏ธ", | |
) | |
# Main | |
st.write("# CityDreamer Minimal Demo ๐๏ธ") | |
with open("README.md", "r") as f: | |
markdown = f.read() | |
st.markdown(markdown[markdown.rfind("---") :]) | |
imgbox = st.empty() | |
# Sidebar | |
st.sidebar.header("CityDreamer Settings") | |
radius = st.sidebar.slider("Camera Radius (m)", 128, 512, 320, 5) | |
altitude = st.sidebar.slider("Camera Altitude (m)", 256, 512, 384, 5) | |
azimuth = st.sidebar.slider("Camera Azimuth (ยฐ)", 0, 360, 180, 5) | |
if st.sidebar.button("Generate", type="primary"): | |
img = get_generated_city(radius, altitude, azimuth) | |
imgbox.image(img, caption="CityDreamer Generation") | |
if __name__ == "__main__": | |
LOGGER.info("Setting up runtime environment...") | |
setup_runtime_env() | |
fgm, bgm = get_models() | |
LOGGER.info("Starting the main application...") | |
main(fgm, bgm) | |