Update app.py

app.py

@@ -52,21 +52,22 @@ def predict(text):
   return pipe(text)[0]["translation_text"]
  
 def main():
-    torch.cuda.init()
-    if torch.cuda.is_initialized():
-        ini = "True1"
-    else:
-        ini = "False1"
-    if torch.cuda.is_available():
-        result = "True2"
-    else:
-        result = "False2"
+    #torch.cuda.init()
+    #if torch.cuda.is_initialized():
+    #    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 = os.system('nvidia-smi')
     iface = gr.Interface(
       fn=predict, 
       inputs='text',
       outputs='text',
-      examples=[[f'{ini}   {result}']]
+      examples=[[f'{result}']]
     )
     
     iface.launch()