Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 import numpy as np
 import os
 
+import torch
+print(f"==> {torch.__version__}")
+print(f"==> {torch.version.cuda}")
+
 try:
     from train import *
     print('==> simple-knn & diff-gaussian-rasterization already installed!')