Update app.py

app.py

@@ -44,8 +44,10 @@ def install_cuda_toolkit():
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     print("==> finished installation")
-    
+
+print("installing cuda toolkit")
 install_cuda_toolkit()
+peinr("finished")
 
 print(os.environ["CUDA_HOME"])