codeslake commited on
Commit
709fd3c
1 Parent(s): 123ab44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -24,9 +24,8 @@ os.makedirs(Ref_path)
24
  os.makedirs(Ref_path_T)
25
  os.makedirs('result')
26
 
27
- os.system("wget https://www.dropbox.com/s/xv6inxwy0so4ni0/LR.png -O {}/0000.png".format(LR_path))
28
- os.system("wget https://www.dropbox.com/s/abydd1oczs1163l/Ref.png -O {}/0000.png".format(Ref_path))
29
- os.system("wget https://www.dropbox.com/s/abydd1oczs1163l/Ref.png -O {}/0000.png".format(Ref_path_T))
30
 
31
  def resize(img):
32
  max_side = 1024
@@ -43,9 +42,9 @@ def inference(LR, Ref):
43
  #LR = resize(LR)
44
  #Ref = resize(Ref)
45
 
46
- #LR.save(os.path.join(LR_path, '0000.png'))
47
- #Ref.save(os.path.join(Ref_path, '0000.png'))
48
- #Ref.save(os.path.join(Ref_path_T, '0000.png'))
49
 
50
  os.system("python -B run.py \
51
  --mode amp_RefVSR_small_MFID_8K \
@@ -73,6 +72,6 @@ article = "<p style='text-align: center'><b>To check the full capability of the
73
  #Image.open('LR.png').save(os.path.join(LR_path, '0000.png'))
74
  #Image.open('LR.png')
75
 
76
- examples=[[[os.path.join(LR_path, '0000.png')], [os.path.join(Ref_path, '0000.png')]]]
77
 
78
  gr.Interface(inference,[gr.inputs.Image(type="pil"), gr.inputs.Image(type="pil")],gr.outputs.Image(type="file"),title=title,description=description,article=article,theme ="peach",examples=examples).launch(enable_queue=True)
24
  os.makedirs(Ref_path_T)
25
  os.makedirs('result')
26
 
27
+ os.system("wget https://www.dropbox.com/s/xv6inxwy0so4ni0/LR.png -O LR.png")
28
+ os.system("wget https://www.dropbox.com/s/abydd1oczs1163l/Ref.png -O Ref.png")
 
29
 
30
  def resize(img):
31
  max_side = 1024
42
  #LR = resize(LR)
43
  #Ref = resize(Ref)
44
 
45
+ LR.save(os.path.join(LR_path, '0000.png'))
46
+ Ref.save(os.path.join(Ref_path, '0000.png'))
47
+ Ref.save(os.path.join(Ref_path_T, '0000.png'))
48
 
49
  os.system("python -B run.py \
50
  --mode amp_RefVSR_small_MFID_8K \
72
  #Image.open('LR.png').save(os.path.join(LR_path, '0000.png'))
73
  #Image.open('LR.png')
74
 
75
+ examples=[['LR.png'], ['Ref.png']]
76
 
77
  gr.Interface(inference,[gr.inputs.Image(type="pil"), gr.inputs.Image(type="pil")],gr.outputs.Image(type="file"),title=title,description=description,article=article,theme ="peach",examples=examples).launch(enable_queue=True)