manhkhanhUIT commited on
Commit
b09b653
1 Parent(s): 2736057

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, colorizaition_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",