Update app.py

app.py

@@ -7,16 +7,14 @@ import os
 #os.system('pip install python-dotenv')
 # In[1]:
 #Use GPU
-gpu_info = get_ipython().getoutput('nvidia-smi')
-gpu_info = '\n'.join(gpu_info)
-if gpu_info.find('failed') >= 0:
-  print('Not connected to a GPU')
-  is_gpu=False
+import subprocess
+gpu_info = subprocess.getoutput('nvidia-smi')
+if 'failed' in gpu_info:
+    print('Not connected to a GPU')
+    is_gpu = False
 else:
-  print(gpu_info)
-  is_gpu=True
-
-#is_gpu=False
+    print(gpu_info)
+    is_gpu = True
 # In[2]:
 # In[3]:
 import os