Spaces:
Sleeping
Sleeping
Commit ·
438cde9
1
Parent(s): a77d74a
model server url updated
Browse files- core.py +1 -1
- parameters.py +2 -1
core.py
CHANGED
|
@@ -368,7 +368,7 @@ def regenerate_data(
|
|
| 368 |
|
| 369 |
try:
|
| 370 |
data = iowrapper.get_valid_post_response(
|
| 371 |
-
url="
|
| 372 |
payload=payload,
|
| 373 |
)
|
| 374 |
image_data = io.BytesIO(base64.b64decode(data["image"]))
|
|
|
|
| 368 |
|
| 369 |
try:
|
| 370 |
data = iowrapper.get_valid_post_response(
|
| 371 |
+
url=f"{parameters.MODEL_SERVER_URL}generate_image",
|
| 372 |
payload=payload,
|
| 373 |
)
|
| 374 |
image_data = io.BytesIO(base64.b64decode(data["image"]))
|
parameters.py
CHANGED
|
@@ -10,4 +10,5 @@ os.environ["AWS_SECRET_ACCESS_KEY"] = os.getenv("AWS_SECRET_ACCESS_KEY")
|
|
| 10 |
os.environ["S3_BUCKET_NAME"] = os.getenv("AWS_BUCKET")
|
| 11 |
VISUAL_CHOICES = ["DARK", "FLUX", "GHIBLI_COMIC"]
|
| 12 |
MAX_TRIES = os.getenv("MAX_TRIES")
|
| 13 |
-
OPEN_AI_API_KEY = os.getenv("OPEN_AI_KEY")
|
|
|
|
|
|
| 10 |
os.environ["S3_BUCKET_NAME"] = os.getenv("AWS_BUCKET")
|
| 11 |
VISUAL_CHOICES = ["DARK", "FLUX", "GHIBLI_COMIC"]
|
| 12 |
MAX_TRIES = os.getenv("MAX_TRIES")
|
| 13 |
+
OPEN_AI_API_KEY = os.getenv("OPEN_AI_KEY")
|
| 14 |
+
MODEL_SERVER_URL = os.getenv("MODEL_SERVER_URL")
|