itacaiunas
commited on
Commit
·
2c7ede3
1
Parent(s):
cc83d1f
Update app.py
Browse files
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)")
|