debugging
Browse files
app.py
CHANGED
@@ -275,6 +275,7 @@ cleanup_thread.start()
|
|
275 |
|
276 |
@spaces.GPU
|
277 |
def preload():
|
|
|
278 |
pipeline.cuda()
|
279 |
try:
|
280 |
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|
@@ -285,6 +286,5 @@ def preload():
|
|
285 |
# Launch the Gradio app
|
286 |
if __name__ == "__main__":
|
287 |
global pipeline
|
288 |
-
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
289 |
preload()
|
290 |
demo.launch()
|
|
|
275 |
|
276 |
@spaces.GPU
|
277 |
def preload():
|
278 |
+
pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
|
279 |
pipeline.cuda()
|
280 |
try:
|
281 |
pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))) # Preload rembg
|
|
|
286 |
# Launch the Gradio app
|
287 |
if __name__ == "__main__":
|
288 |
global pipeline
|
|
|
289 |
preload()
|
290 |
demo.launch()
|