Spaces:

owaiskaifi
/

qr-ai

Runtime error

owaiskaifi commited on Jun 22, 2023

Commit

7171058

•

1 Parent(s): ff05e00

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ from diffusers.utils import load_image
 app = Flask(__name__)
-# controlnet = ControlNetModel.from_pretrained("DionTimmer/controlnet_qrcode-control_v1p_sd15",
-#                                              torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",

 app = Flask(__name__)
+controlnet = ControlNetModel.from_pretrained("DionTimmer/controlnet_qrcode-control_v1p_sd15",
+                                             torch_dtype=torch.float16, local_files_only=False)
 pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",