Spaces:
Build error
Build error
Commit
·
b09b653
1
Parent(s):
2736057
Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ def run_cmd(command):
|
|
126 |
print("Process interrupted")
|
127 |
sys.exit(1)
|
128 |
|
129 |
-
def run(image,Restoration_mode,
|
130 |
if Restoration_mode == "BOPBTL":
|
131 |
if os.path.isdir("Temp"):
|
132 |
shutil.rmtree("Temp")
|
@@ -156,7 +156,7 @@ def run(image,Restoration_mode, colorizaition_mode):
|
|
156 |
|
157 |
return result_colorization
|
158 |
|
159 |
-
examples = [['example/1.jpeg'],['example/2.jpg'],['example/3.jpg'],['example/4.jpg']]
|
160 |
iface = gr.Interface(run,
|
161 |
[gr.inputs.Image(),gr.inputs.Radio(["BOPBTL", "Pix2Pix"]),gr.inputs.Radio(["Deoldify", "Pix2Pix Resnet 9block","Pix2Pix Unet 256"])],
|
162 |
outputs="image",
|
|
|
126 |
print("Process interrupted")
|
127 |
sys.exit(1)
|
128 |
|
129 |
+
def run(image,Restoration_mode, Colorizaition_mode):
|
130 |
if Restoration_mode == "BOPBTL":
|
131 |
if os.path.isdir("Temp"):
|
132 |
shutil.rmtree("Temp")
|
|
|
156 |
|
157 |
return result_colorization
|
158 |
|
159 |
+
examples = [['example/1.jpeg',"BOPBTL","Deoldify"],['example/2.jpg',"BOPBTL","Deoldify"],['example/3.jpg',"BOPBTL","Deoldify"],['example/4.jpg',"BOPBTL","Deoldify"]]
|
160 |
iface = gr.Interface(run,
|
161 |
[gr.inputs.Image(),gr.inputs.Radio(["BOPBTL", "Pix2Pix"]),gr.inputs.Radio(["Deoldify", "Pix2Pix Resnet 9block","Pix2Pix Unet 256"])],
|
162 |
outputs="image",
|