feng2022 commited on
Commit
9e99e2a
1 Parent(s): b3b8c9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -58,10 +58,10 @@ def main():
58
  # ini = "True1"
59
  #else:
60
  # ini = "False1"
61
- #if torch.cuda.is_available():
62
- # result = "True2"
63
- # else:
64
- # result = "False2"
65
  #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
66
  result = subprocess.check_output(['nvidia-smi'])
67
  iface = gr.Interface(
 
58
  # ini = "True1"
59
  #else:
60
  # ini = "False1"
61
+ if torch.cuda.is_available():
62
+ result = "True2"
63
+ else:
64
+ result = "False2"
65
  #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
66
  result = subprocess.check_output(['nvidia-smi'])
67
  iface = gr.Interface(