Spaces:
Runtime error
Runtime error
File size: 4,516 Bytes
05fd390 5a2793f 4efa9a6 e384b8f a267f7a 05fd390 9db9711 034d231 3d8396f 05fd390 8e4a774 8f17adb 05fd390 3d8396f 3775564 b0a2e4f 6debf2b b0a2e4f 3775564 b0a2e4f 9aaa365 9db9711 8e4a774 3d8396f 8e4a774 ad93de6 b0a2e4f 05fd390 3d8396f |
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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 |
import gradio as gr
import torch
import os
import shutil
import requests
import subprocess
from subprocess import getoutput
from huggingface_hub import login, HfFileSystem, snapshot_download, HfApi, create_repo
from pathlib import Path
from PIL import Image
from app_train import create_training_demo
from sdxl.app_inference import create_inference_demo
from depthgltf.app_visualisations import create_visual_demo
from transformers import DPTFeatureExtractor, DPTForDepthEstimation
import numpy as np
import open3d as o3d
css="""
#col-container {max-width: 780px; margin-left: auto; margin-right: auto;}
#upl-dataset-group {background-color: none!important;}
div#warning-ready {
background-color: #ecfdf5;
padding: 0 10px 5px;
margin: 20px 0;
}
div#warning-ready > .gr-prose > h2, div#warning-ready > .gr-prose > p {
color: #057857!important;
}
div#warning-duplicate {
background-color: #ebf5ff;
padding: 0 10px 5px;
margin: 20px 0;
}
div#warning-duplicate > .gr-prose > h2, div#warning-duplicate > .gr-prose > p {
color: #0f4592!important;
}
div#warning-duplicate strong {
color: #0f4592;
}
p.actions {
display: flex;
align-items: center;
margin: 20px 0;
}
div#warning-duplicate .actions a {
display: inline-block;
margin-right: 10px;
}
div#warning-setgpu {
background-color: #fff4eb;
padding: 0 10px 5px;
margin: 20px 0;
}
div#warning-setgpu > .gr-prose > h2, div#warning-setgpu > .gr-prose > p {
color: #92220f!important;
}
div#warning-setgpu a, div#warning-setgpu b {
color: #91230f;
}
div#warning-setgpu p.actions > a {
display: inline-block;
background: #1f1f23;
border-radius: 40px;
padding: 6px 24px;
color: antiquewhite;
text-decoration: none;
font-weight: 600;
font-size: 1.2em;
}
button#load-dataset-btn{
min-height: 60px;
}
"""
theme = gr.themes.Soft(
primary_hue="teal",
secondary_hue="gray",
).set(
body_text_color_dark='*neutral_800',
background_fill_primary_dark='*neutral_50',
background_fill_secondary_dark='*neutral_50',
border_color_accent_dark='*primary_300',
border_color_primary_dark='*neutral_200',
color_accent_soft_dark='*neutral_50',
link_text_color_dark='*secondary_600',
link_text_color_active_dark='*secondary_600',
link_text_color_hover_dark='*secondary_700',
link_text_color_visited_dark='*secondary_500',
code_background_fill_dark='*neutral_100',
shadow_spread_dark='6px',
block_background_fill_dark='white',
block_label_background_fill_dark='*primary_100',
block_label_text_color_dark='*primary_500',
block_title_text_color_dark='*primary_500',
checkbox_background_color_dark='*background_fill_primary',
checkbox_background_color_selected_dark='*primary_600',
checkbox_border_color_dark='*neutral_100',
checkbox_border_color_focus_dark='*primary_500',
checkbox_border_color_hover_dark='*neutral_300',
checkbox_border_color_selected_dark='*primary_600',
checkbox_label_background_fill_selected_dark='*primary_500',
checkbox_label_text_color_selected_dark='white',
error_background_fill_dark='#fef2f2',
error_border_color_dark='#b91c1c',
error_text_color_dark='#b91c1c',
error_icon_color_dark='#b91c1c',
input_background_fill_dark='white',
input_background_fill_focus_dark='*secondary_500',
input_border_color_dark='*neutral_50',
input_border_color_focus_dark='*secondary_300',
input_placeholder_color_dark='*neutral_400',
slider_color_dark='*primary_500',
stat_background_fill_dark='*primary_300',
table_border_color_dark='*neutral_300',
table_even_background_fill_dark='white',
table_odd_background_fill_dark='*neutral_50',
button_primary_background_fill_dark='*primary_500',
button_primary_background_fill_hover_dark='*primary_400',
button_primary_border_color_dark='*primary_00',
button_secondary_background_fill_dark='whiite',
button_secondary_background_fill_hover_dark='*neutral_100',
button_secondary_border_color_dark='*neutral_200',
button_secondary_text_color_dark='*neutral_800'
)
with gr.Blocks(theme=theme, css=css) as demo:
gr.Markdown("SUTD x SUNS Shop Design Generator")
with gr.Tab("Training"):
create_training_demo()
with gr.Tab("Generation"):
create_inference_demo()
#create_visual_demo();
with gr.Tab("Visualisation"):
create_visual_demo();
demo.queue().launch(debug=True, share=True)
|