Update app.py

app.py

@@ -24,7 +24,7 @@ import datetime
 app = FastAPI()
 
 
-# subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl'))
+subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl'))
 
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
@@ -44,7 +44,7 @@ model = TSR.from_pretrained(
     config_name="config.yaml",
     weight_name="model.ckpt",
 )
-model.renderer.set_chunk_size(131072)
+# model.renderer.set_chunk_size(131072)
 model.to(device)
 
 rembg_session = rembg.new_session()
@@ -126,7 +126,7 @@ def gen_pos_prompt(text):
 
 def generate_image_from_text(pos_prompt, seed):
   new_prompt = gen_pos_prompt(pos_prompt)
-  print(new_prompt)
+  # print(new_prompt)
   # neg_prompt = '''Detailed, complex textures, intricate patterns, realistic lighting, high contrast, reflections, fuzzy surface, realistic proportions, photographic quality, vibrant colors, detailed background, shadows, disfigured, deformed, ugly, multiple, duplicate.'''
   # neg_prompt = '''Complex textures, intricate patterns, realistic lighting, high contrast, reflections, fuzzy surface, photographic quality, vibrant colors, detailed background, shadows, disfigured, deformed, ugly, multiple, duplicate.'''
   neg_prompt = '''Complex patterns, realistic lighting, high contrast, reflections, fuzzy, photographic, vibrant, detailed, shadows, disfigured, duplicate.'''
@@ -137,8 +137,8 @@ def generate_image_from_text(pos_prompt, seed):
                             'negativeText': neg_prompt},
       'imageGenerationConfig': {"cfgScale":8,
                                 "seed":int(seed),
-                                "width":768,
-                                "height":768,
+                                "width":512,
+                                "height":512,
                                 "numberOfImages":1
                                 }
   }