Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
 import os
+import torch
 import shutil
 import spaces