Update app.py

app.py

@@ -25,6 +25,7 @@ import torch.nn.functional as F
 from torchvision import transforms
 
 import subprocess
+import spaces
 
 device = "cuda" if torch.cuda.is_available() else "cpu"