itacaiunas commited on
Commit
2c7ede3
·
1 Parent(s): cc83d1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -11,9 +11,12 @@ from streamlit_drawable_canvas import st_canvas
11
  from io import BytesIO
12
  from copy import deepcopy
13
 
 
 
14
  from src.core import process_inpaint
15
 
16
 
 
17
  def image_download_button(pil_image, filename: str, fmt: str, label="Download"):
18
  if fmt not in ["jpg", "png"]:
19
  raise Exception(f"Unknown image format (Available: {fmt} - case sensitive)")
 
11
  from io import BytesIO
12
  from copy import deepcopy
13
 
14
+ import sys
15
+ sys.path.insert(0, '/tree/main/src')
16
  from src.core import process_inpaint
17
 
18
 
19
+
20
  def image_download_button(pil_image, filename: str, fmt: str, label="Download"):
21
  if fmt not in ["jpg", "png"]:
22
  raise Exception(f"Unknown image format (Available: {fmt} - case sensitive)")