Update app.py

app.py

@@ -1,5 +1,5 @@
-import torch
-from torch import autocast
+import pytorch
+from pytorch import autocast
 from diffusers import StableDiffusionPipeline
 from datasets import load_dataset
 from PIL import Image