Update app.py

app.py

@@ -58,12 +58,9 @@ def main():
     #    ini = "True1"
     #else:
     #    ini = "False1"
-    if torch.cuda.is_available():
-        result = "True2"
-    else:
-        result = "False2"
-    #load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
     #result = subprocess.check_output(['nvidia-smi'])
+    load_model("stylegan2-ffhq-config-f","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f")
+    
     iface = gr.Interface(
       fn=predict, 
       inputs='text',