Update app.py

app.py

@@ -4,7 +4,7 @@ import math
 import os
 from contextlib import nullcontext
 import random
-
+import torch
 
 import PIL
 from accelerate import Accelerator