Commit
•
52f84ce
1
Parent(s):
da0c7e5
Update handler.py
Browse files- 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 = "
|
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,
|