Update app.py

app.py

@@ -15,7 +15,8 @@ import cv2
 from matplotlib import pyplot as plt
 from torchvision import transforms
 from diffusers import DiffusionPipeline
-from diffusers.utils import torch_device
+import torch
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 pipe = DiffusionPipeline.from_pretrained(
     "Fantasy-Studio/Paint-by-Example",
     torch_dtype=torch.float16,