Spaces:
Starting
on
A10G
Starting
on
A10G
File size: 1,398 Bytes
c2164fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
# -*- 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:26:57
# @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():
pass
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)
|