Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import os
 import sys
@@ -8,6 +9,10 @@ import numpy as np
 from PIL import Image
 
 import torch
+
+
+print(f'torch version:{torch.__version__}')
+
 import torch.utils.checkpoint
 from pytorch_lightning import seed_everything
 from diffusers import AutoencoderKL, DDPMScheduler