Update app.py

app.py

@@ -12,9 +12,9 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 # Загружаем модель
 model = TSR.from_pretrained(
-    "stabi/TripoSR",
-    config_name=" config.yaml",
-    weight_name=" model.ckpt",
+    "stabilityai/TripoSR",
+    config_name="config.yaml",
+    weight_name="model.ckpt",
 )
 model.renderer.set_chunk_size(131072)
 model.to(device)
@@ -25,35 +25,38 @@ def check_input_image(input_image):
     if input_image is None:
         raise gr.Error("No image uploaded!")
 
+
 # Функция обработки изображения
 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
         image = Image.fromarray((image * 255.0).astype(np.uint8))
-    return image
+        return image
 
     if do_remove_background:
         image = input_image.convert("RGB")
-        image = removebackground(image)
+        image = remove_background(image)
         image = resize_foreground(image, foreground_ratio)
-        image = fillbackground(image)
+        image = fill_background(image)
     else:
         image = input_image
         if image.mode == "RGBA":
-            image = fillbackground(image)
+            image = fill_background(image)
     return image
 
+
 # Функция генерации 3D модели
 def generate(image):
     time.sleep(3)  # Эмуляция времени обработки
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes)[0]
     mesh = to_gradio_3d_orientation(mesh)
-    mesh_path2 = tempfile.Namedtemporaryfile(delete=False)
+    mesh_path2 = tempfile.NamedTemporaryFile(suffix=".glb", delete=False)
     mesh.export(mesh_path2.name)
     return mesh_path2.name
 
+
 # Функции для управления лоадерами
 def start_loading_processed():
     return "<div id='loading-bar-processed' style='display: block;'></div>"
@@ -67,6 +70,7 @@ def start_loading_glb():
 def stop_loading_glb():
     return "<div id='loading-bar-glb' style='display: none;'></div>"
 
+
 # Настройка темы и CSS
 class CustomTheme(gr.themes.Base):
     def __init__(self):
@@ -76,10 +80,11 @@ class CustomTheme(gr.themes.Base):
         self.background_fill_secondary = "#191a1e"
         self.background_fill_tertiary = "#191a1e"
         self.text_color_primary = "#FFFFFF"
-        self.textcolor_secondary = "#FFFFFF"
-        self.textcolor_tertiary = "#FFFFFF"
+        self.text_color_secondary = "#FFFFFF"
+        self.text_color_tertiary = "#FFFFFF"
         self.input_background_fill = "#191a1e"
-        self.inputtext_color = "#FFFFFF"
+        self.input_text_color = "#FFFFFF"
+
 
 css = """
 /* Скрываем системный лоадер Gradio */
@@ -203,4 +208,4 @@ demo.launch(
     server_name="0.0.0.0",
     server_port=7860,
     share=True,
-)
+)