Spaces:
Runtime error
Runtime error
lionelgarnier
commited on
Commit
·
4b713ff
1
Parent(s):
00ca2fb
fix trellis pipeline
Browse files
app.py
CHANGED
@@ -63,7 +63,11 @@ def end_session(req: gr.Request):
|
|
63 |
shutil.rmtree(user_dir)
|
64 |
|
65 |
def preprocess_image(image: Image.Image) -> Image.Image:
|
66 |
-
|
|
|
|
|
|
|
|
|
67 |
return processed_image
|
68 |
|
69 |
|
|
|
63 |
shutil.rmtree(user_dir)
|
64 |
|
65 |
def preprocess_image(image: Image.Image) -> Image.Image:
|
66 |
+
trellis = get_trellis_pipeline()
|
67 |
+
if trellis is None:
|
68 |
+
# If the pipeline is not loaded, just return the original image
|
69 |
+
return image
|
70 |
+
processed_image = trellis.preprocess_image(image)
|
71 |
return processed_image
|
72 |
|
73 |
|