vertxlabs
/

controlnet_qrcode-control_v11p_v1

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

Update handler.py

#3

by realcodeninja - opened Jul 13, 2023

base: refs/heads/main

←

from: refs/pr/3

Discussion Files changed

Files changed (1) hide show

handler.py +3 -3

handler.py CHANGED Viewed

@@ -1,5 +1,5 @@
 # handler.py
 from PIL import Image
 from diffusers import (
     StableDiffusionControlNetImg2ImgPipeline,
@@ -12,7 +12,7 @@ import openai
 from io import BytesIO
 import base64
 import qrcode
 class EndpointHandler:
     def __init__(
@@ -60,7 +60,7 @@ class EndpointHandler:
         height,
         num_inference_steps,
     ):
-        openai.api_key = "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"
         qr = qrcode.QRCode(
             version=1,

 # handler.py
+from dotenv import load_dotenv
 from PIL import Image
 from diffusers import (
     StableDiffusionControlNetImg2ImgPipeline,
 from io import BytesIO
 import base64
 import qrcode
+load_dotenv()
 class EndpointHandler:
     def __init__(
         height,
         num_inference_steps,
     ):
+        openai.api_key = os.getenv("OPENAI_KEY")
         qr = qrcode.QRCode(
             version=1,