umang-immersfy commited on
Commit
438cde9
·
1 Parent(s): a77d74a

model server url updated

Browse files
Files changed (2) hide show
  1. core.py +1 -1
  2. 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="http://10.100.111.13:4389/generate_image",
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")