Update app.py

app.py

@@ -80,6 +80,7 @@ md = "![](https://drive.google.com/uc?id=1WBqYsKRwn_78A72MJBrk643l3-gfAssP) \n"
      "This project is licensed under the terms of the MIT license. \n"
 
 if __name__ =='__main__':
+    torch.cuda.init()
     print(f"Is CUDA available: {torch.cuda.is_available()}")
     # True
     print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")