realcodeninja commited on
Commit
52f84ce
1 Parent(s): da0c7e5

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -2
handler.py CHANGED
@@ -1,6 +1,7 @@
1
  # handler.py
2
-
3
  from PIL import Image
 
4
  from diffusers import (
5
  StableDiffusionControlNetImg2ImgPipeline,
6
  ControlNetModel,
@@ -13,6 +14,7 @@ from io import BytesIO
13
  import base64
14
  import qrcode
15
 
 
16
 
17
  class EndpointHandler:
18
  def __init__(
@@ -60,7 +62,7 @@ class EndpointHandler:
60
  height,
61
  num_inference_steps,
62
  ):
63
- openai.api_key = "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"
64
 
65
  qr = qrcode.QRCode(
66
  version=1,
 
1
  # handler.py
2
+ from dotenv import load_dotenv
3
  from PIL import Image
4
+ import os
5
  from diffusers import (
6
  StableDiffusionControlNetImg2ImgPipeline,
7
  ControlNetModel,
 
14
  import base64
15
  import qrcode
16
 
17
+ load_dotenv()
18
 
19
  class EndpointHandler:
20
  def __init__(
 
62
  height,
63
  num_inference_steps,
64
  ):
65
+ openai.api_key = os.getenv("OPENAI_KEY")
66
 
67
  qr = qrcode.QRCode(
68
  version=1,