lionelgarnier commited on
Commit
4b713ff
·
1 Parent(s): 00ca2fb

fix trellis pipeline

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- processed_image = _trellis_pipeline.preprocess_image(image)
 
 
 
 
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