Update app.py

app.py

@@ -13,25 +13,19 @@ from functools import partial
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
-# HF_TOKEN = os.getenv("HF_TOKEN")
-
-HEADER = """
-"""
-
 if torch.cuda.is_available():
     device = "cuda:0"
 else:
     device = "cpu"
 
 d = os.environ.get("DEVICE", None)
-if d!= None:
+if d != None:
     device = d 
 
 model = TSR.from_pretrained(
     "stabilityai/TripoSR",
     config_name="config.yaml",
     weight_name="model.ckpt",
-#    token=HF_TOKEN
 )
 model.renderer.set_chunk_size(131072)
 model.to(device)
@@ -74,28 +68,21 @@ def generate(image):
     return mesh_path.name, mesh_path2.name
 
 
-def run_example(image_pil):
-    preprocessed = preprocess(image_pil, False, 0.9)
-    mesh_name, mesn_name2 = generate(preprocessed)
-    return preprocessed, mesh_name, mesn_name2
-
-
 class CustomTheme(gr.themes.Base):
     def __init__(self):
         super().__init__()
         self.primary_hue = "#191a1e"
-        self.background_fill_primary = "#191a1e"
-        self.background_fill_secondary = "#191a1e"
-        self.background_fill_tertiary = "#191a1e"
+        self.background_fill_primary = "#1a1a1a"
+        self.background_fill_secondary = "#1a1a1a"
+        self.background_fill_tertiary = "#1a1a1a"
         self.text_color_primary = "#FFFFFF"
         self.text_color_secondary = "#FFFFFF"
         self.text_color_tertiary = "#FFFFFF"
-        self.input_background_fill = "#191a1e"
+        self.input_background_fill = "#1a1a1a"
         self.input_text_color = "#FFFFFF"
 
 
 css = """
-/* Скрываем нижний колонтитул */
 footer {
     visibility: hidden;
     height: 0;
@@ -104,29 +91,24 @@ footer {
     overflow: hidden;
 }
 
-/* Применяем шрифты */
-@import url('https://fonts.googleapis.com/css2?family=Poppins:wght@400;500;700&display=swap');
 body, input, button, textarea, select, .gr-button {
     font-family: 'Poppins', sans-serif;
-    background-color: #191a1e !important;
+    background-color: #1a1a1a !important;
     color: #FFFFFF;
 }
 
-/* Настройки заголовков */
 h1, h2, h3, h4, h5, h6 {
     font-family: 'Poppins', sans-serif;
     font-weight: 700;
     color: #FFFFFF;
 }
 
-/* Стиль для текстовых полей и кнопок */
 input[type="text"], textarea {
-    background-color: #191a1e !important;
+    background-color: #1a1a1a !important;
     color: #FFFFFF;
     border: 1px solid #FFFFFF;
 }
 
-/* Цвет кнопки Generate */
 .generate-button {
     background-color: #5271FF !important;
     color: #FFFFFF !important;
@@ -135,18 +117,18 @@ input[type="text"], textarea {
 }
 
 .generate-button:hover {
-    background-color: #405BBF !important; /* Цвет при наведении */
+    background-color: #405BBF !important;
 }
 
 gr-image-upload {
-    background-color: transparent !important; /* Прозрачный фон */
-    border: 1px dashed #FFFFFF !important; /* Белая пунктирная рамка */
-    position: relative; /* Для позиционирования текста */
+    background-color: #1a1a1a !important;
+    border: 1px dashed #FFFFFF !important;
+    position: relative;
 }
 
 .gr-image-upload .dropzone::before {
     content: 'Drop Image Here';
-    color: #FFFFFF; /* Белый цвет текста */
+    color: #FFFFFF;
     font-size: 1rem;
     font-weight: bold;
     text-align: center;
@@ -155,12 +137,11 @@ gr-image-upload {
     left: 50%;
     transform: translate(-50%, -50%);
 }
-/* Скрытие стандартной иконки загрузки */
+
 .gr-image-upload .dropzone .file-upload {
     display: none !important;
 }
 
-/* Выравнивание элементов */
 .drop-image-container {
     display: flex;
     flex-direction: column;
@@ -182,12 +163,11 @@ gr-image-upload {
     margin-right: auto;
 }
 """
+
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     with gr.Column(elem_id="main-container"):
-        # Заголовок
         gr.Markdown("**Upload and Process Your Image**", elem_classes="prompt-text")
 
-        # Размещение изображений
         with gr.Row(elem_id="image-row"):
             input_image = gr.Image(
                 label="Upload Image",
@@ -206,7 +186,6 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
                 height=300,
             )
 
-        # Слайдер для параметров
         with gr.Row(elem_id="slider-row"):
             foreground_ratio = gr.Slider(
                 label="Foreground Ratio",
@@ -216,7 +195,6 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
                 step=0.05,
             )
 
-        # Настройки и кнопка генерации
         with gr.Row(elem_id="settings-row"):
             do_remove_background = gr.Checkbox(
                 label="Remove Background",
@@ -229,7 +207,6 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
                 elem_classes="generate-button",
             )
 
-        # Результ��ты генерации
         with gr.Row(elem_id="output-model-container"):
             with gr.Tab("OBJ Model"):
                 output_model = gr.Model3D(
@@ -242,7 +219,6 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
                     interactive=False,
                 )
 
-        # Привязка событий
         submit.click(
             fn=check_input_image,
             inputs=[input_image],
@@ -257,6 +233,5 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
             outputs=[output_model, output_model2]
         )
 
-# Запуск интерфейса
 demo.queue(max_size=10)
 demo.launch()