Update app.py

app.py

@@ -10,7 +10,7 @@ import torch
 import numpy as np
 import imageio
 from easydict import EasyDict as edict
-from PIL import Image
+from PIL import Image, ImageOps
 from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
@@ -35,16 +35,6 @@ import cv2
 import os
 import random
 
-js_func = """
-function refresh() {
-    const url = new URL(window.location);
-
-    if (url.searchParams.get('__theme') !== 'dark') {
-        url.searchParams.set('__theme', 'dark');
-        window.location.href = url.href;
-    }
-}
-"""
 
 style_list = [
     {
@@ -145,6 +135,7 @@ def preprocess_image(image: Image.Image,
     ratio = np.sqrt(1024. * 1024. / (width * height))
     new_width, new_height = int(width * ratio), int(height * ratio)
     image = image['composite'].resize((new_width, new_height))
+    image = ImageOps.invert(image)
 
     print("image:",type(image))
 
@@ -178,7 +169,7 @@ def preprocess_images(images: List[Tuple[Image.Image, str]]) -> List[Image.Image
     """
     images = [image[0] for image in images]
     processed_images = [pipeline.preprocess_image(image) for image in images]
-    return processed_images, False
+    return processed_images
 
 
 def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
@@ -260,6 +251,7 @@ def image_to_3d(
         str: The path to the video of the 3D model.
     """
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
+    print("image 2", type(image))
     if not is_multiimage:
         outputs = pipeline.run(
             image,
@@ -379,10 +371,10 @@ def split_image(image: Image.Image) -> List[Image.Image]:
     return [preprocess_image(image) for image in images]
 
 
-with gr.Blocks(delete_cache=(600, 600), js=js_func) as demo:
+with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
     ## Sketch to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
-    * draw or upload a sketch and click "Generate" to create a 3D asset. 
+    * 🎨🖌️ draw or upload a sketch and click "Generate" to create a 3D asset. 
     """)
     
     with gr.Row():
@@ -444,7 +436,7 @@ with gr.Blocks(delete_cache=(600, 600), js=js_func) as demo:
                         slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
                     multiimage_algo = gr.Radio(["stochastic", "multidiffusion"], label="Multi-image Algorithm", value="stochastic")
                
-            generate_btn = gr.Button("Generate")
+            #generate_btn = gr.Button("Generate")
             
             with gr.Accordion(label="GLB Extraction Settings", open=False):
                 mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)