Update app.py

app.py

@@ -12,13 +12,14 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 # Загружаем модель
 model = TSR.from_pretrained(
-    "stabilityai/TripoSR",
-    config_name="config.yaml",
-    weight_name="model.ckpt",
+    "stabi/TripoSR",
+    config_name=" config.yaml",
+    weight_name=" model.ckpt",
 )
 model.renderer.set_chunk_size(131072)
 model.to(device)
 
+
 # Функция для проверки изображения
 def check_input_image(input_image):
     if input_image is None:
@@ -29,17 +30,18 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
     def fill_background(image):
         image = np.array(image).astype(np.float32) / 255.0
         image = image[:, :, :3] * image[:, :, 3:4] + (1 - image[:, :, 3:4]) * 0.5
-        return Image.fromarray((image * 255.0).astype(np.uint8))
+        image = Image.fromarray((image * 255.0).astype(np.uint8))
+    return image
 
     if do_remove_background:
         image = input_image.convert("RGB")
-        image = remove_background(image)
+        image = removebackground(image)
         image = resize_foreground(image, foreground_ratio)
-        image = fill_background(image)
+        image = fillbackground(image)
     else:
         image = input_image
         if image.mode == "RGBA":
-            image = fill_background(image)
+            image = fillbackground(image)
     return image
 
 # Функция генерации 3D модели
@@ -48,7 +50,7 @@ def generate(image):
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes)[0]
     mesh = to_gradio_3d_orientation(mesh)
-    mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
+    mesh_path2 = tempfile.Namedtemporaryfile(delete=False)
     mesh.export(mesh_path2.name)
     return mesh_path2.name
 
@@ -74,10 +76,10 @@ class CustomTheme(gr.themes.Base):
         self.background_fill_secondary = "#191a1e"
         self.background_fill_tertiary = "#191a1e"
         self.text_color_primary = "#FFFFFF"
-        self.text_color_secondary = "#FFFFFF"
-        self.text_color_tertiary = "#FFFFFF"
+        self.textcolor_secondary = "#FFFFFF"
+        self.textcolor_tertiary = "#FFFFFF"
         self.input_background_fill = "#191a1e"
-        self.input_text_color = "#FFFFFF"
+        self.inputtext_color = "#FFFFFF"
 
 css = """
 /* Скрываем системный лоадер Gradio */
@@ -90,9 +92,9 @@ footer {
 }
 
 /* Общий стиль */
-body, input, button, textarea, select,.gr-button {
+body, input, button, textarea, select, .gr-button {
     font-family: 'Poppins', sans-serif;
-    background-color: #191a1e!important;
+    background-color: #191a1e !important;
     color: #FFFFFF;
 }
 
@@ -100,6 +102,9 @@ body, input, button, textarea, select,.gr-button {
 #loading-bar-processed, #loading-bar-glb {
     display: none;
     position: absolute;
+    top: 50%;
+    left: 50%;
+    transform: translate(-50%, -50%);
     width: 60px;
     height: 60px;
     border: 8px solid #f3f3f3;
@@ -108,13 +113,6 @@ body, input, button, textarea, select,.gr-button {
     animation: spin 1s linear infinite;
 }
 
-#processed-container #loading-bar-processed,
-.gr-model3d-container #loading-bar-glb {
-    top: 50%;
-    left: 50%;
-    transform: translate(-50%, -50%);
-}
-
 @keyframes spin {
     0% { transform: translate(-50%, -50%) rotate(0deg); }
     100% { transform: translate(-50%, -50%) rotate(360deg); }
@@ -122,7 +120,7 @@ body, input, button, textarea, select,.gr-button {
 
 /* Текст CheckBox в белый */
 .gr-checkbox label {
-    color: #FFFFFF!important;
+    color: #FFFFFF !important;
 }
 """
 
@@ -156,7 +154,7 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
             step=0.05,
         )
         do_remove_background = gr.Checkbox(
-            label="Remove Background",
+            label="Remove Background",  # Текст перекрашен в белый через CSS
             value=True,
         )