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",
|