Fix

app.py

@@ -4,13 +4,14 @@ import torch
 import numpy as np
 from controlnet_aux import HEDdetector
 from PIL import Image
+import os
 
 negative_prompt = ""
 device = 'cpu'
-controlnet = ControlNetModel.from_pretrained("vsanimator/sketchcomplete", torch_dtype=torch.float16).to(device)
+controlnet = ControlNetModel.from_pretrained("vsanimator/sketch-a-sketch").to(device)
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", 
-    controlnet=controlnet, torch_dtype=torch.float16
+    controlnet=controlnet#, torch_dtype=torch.float16
 ).to(device)
 pipe.safety_checker = None
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)

requirements.txt

@@ -1,5 +1,6 @@
-diffusers==0.14.0
-torch==2.0.1
-numpy==1.24.2
-controlnet_aux==0.0.1
-pillow==9.4.0
+gradio
+diffusers
+controlnet_aux
+accelerate
+transformers
+torch