feng2022 commited on
Commit
229a558
·
1 Parent(s): 71264e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -54,13 +54,13 @@ def predict(text):
54
  def main():
55
  torch.cuda.init()
56
  if torch.cuda.is_initialized():
57
- ini = "True"
58
  else:
59
- ini = "False"
60
  if torch.cuda.is_available():
61
- result = "True"
62
  else:
63
- result = "False"
64
  #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
65
  iface = gr.Interface(
66
  fn=predict,
 
54
  def main():
55
  torch.cuda.init()
56
  if torch.cuda.is_initialized():
57
+ ini = "True1"
58
  else:
59
+ ini = "False1"
60
  if torch.cuda.is_available():
61
+ result = "True2"
62
  else:
63
+ result = "False2"
64
  #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
65
  iface = gr.Interface(
66
  fn=predict,