Update app.py

app.py

@@ -51,12 +51,12 @@ def image_create(seed: int, truncation_psi: float):
     return imgs_arr[0]/255
     
 def main():
-    #torch.cuda.init()
-    #if torch.cuda.is_initialized():
-    #    ini = "True1"
-    #else:
-    #    ini = "False1"
-    #result = subprocess.check_output(['nvidia-smi'])
+    torch.cuda.init()
+    if torch.cuda.is_initialized():
+        ini = "True1"
+    else:
+        ini = "False1"
+    result = subprocess.check_output(['nvidia-smi'])
     device = th.device()
     iface = gr.Interface(
           image_create,
@@ -66,7 +66,7 @@ def main():
                     0, 2, step=0.05, default=0.7, label='Truncation psi'),
           ],
           gr.outputs.Image(type='numpy', label='Output'),
-          title=TITLE,
+          title=ini,
           description=DESCRIPTION,
           article=ARTICLE,
           )