feng2022 commited on
Commit
f4f2167
1 Parent(s): abe9d47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -19,10 +19,10 @@ from transformers import pipeline
19
  sys.path.append('.')
20
  sys.path.append('./Time_TravelRephotography')
21
  from argparse import Namespace
22
- #from projector import (
23
- # ProjectorArguments,
24
- # main,
25
- #)
26
  sys.path.insert(0, 'StyleGAN-Human')
27
 
28
  TITLE = 'Time-TravelRephotography'
@@ -73,15 +73,15 @@ def main():
73
  # ini = "False1"
74
  #result = subprocess.check_output(['nvidia-smi'])
75
 
76
- args = parse_args()
77
- device = torch.device(args.device)
78
- load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)
79
 
80
  iface = gr.Interface(
81
  fn=predict,
82
  inputs='text',
83
  outputs='text',
84
- examples=[f'result']
85
  )
86
 
87
  iface.launch()
 
19
  sys.path.append('.')
20
  sys.path.append('./Time_TravelRephotography')
21
  from argparse import Namespace
22
+ from projector import (
23
+ ProjectorArguments,
24
+ main,
25
+ )
26
  sys.path.insert(0, 'StyleGAN-Human')
27
 
28
  TITLE = 'Time-TravelRephotography'
 
73
  # ini = "False1"
74
  #result = subprocess.check_output(['nvidia-smi'])
75
 
76
+ #args = parse_args()
77
+ #device = torch.device(args.device)
78
+ #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",device)
79
 
80
  iface = gr.Interface(
81
  fn=predict,
82
  inputs='text',
83
  outputs='text',
84
+ examples=['result']
85
  )
86
 
87
  iface.launch()