Xuanyou commited on
Commit
3b2c50e
·
verified ·
1 Parent(s): a8c75fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -216,7 +216,7 @@ def spirit_animal_baseline(image_path, num_images = 4):
216
  pose_pil = Image.fromarray(cv2.resize(pose_image, (gen_width, gen_height), interpolation=cv2.INTER_LANCZOS4))
217
 
218
  base64_image = base64.b64encode(cv2.imencode('.jpg', image_rgb)[1]).decode()
219
- api_key = "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"
220
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
221
  payload = {
222
  "model": "gpt-4o-mini",
@@ -297,7 +297,7 @@ def spirit_animal_with_background(image_path, num_images = 4):
297
  pose_pil = Image.fromarray(cv2.resize(pose_image, (gen_width, gen_height), interpolation=cv2.INTER_LANCZOS4))
298
 
299
  base64_image = base64.b64encode(cv2.imencode('.jpg', image_rgb)[1]).decode()
300
- api_key = "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"
301
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
302
  payload = {
303
  "model": "gpt-4o-mini",
@@ -364,7 +364,7 @@ def generate_multiple_animals(image_path, keep_background=True, num_images = 4):
364
  gen_width, gen_height = resize_to_multiple_of_64(gen_width, gen_height)
365
 
366
  base64_image = base64.b64encode(cv2.imencode('.jpg', image_rgb)[1]).decode()
367
- api_key = "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"
368
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
369
  payload = {
370
  "model": "gpt-4o-mini",
 
216
  pose_pil = Image.fromarray(cv2.resize(pose_image, (gen_width, gen_height), interpolation=cv2.INTER_LANCZOS4))
217
 
218
  base64_image = base64.b64encode(cv2.imencode('.jpg', image_rgb)[1]).decode()
219
+ api_key = os.environ.get("OPENAI_API_KEY")
220
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
221
  payload = {
222
  "model": "gpt-4o-mini",
 
297
  pose_pil = Image.fromarray(cv2.resize(pose_image, (gen_width, gen_height), interpolation=cv2.INTER_LANCZOS4))
298
 
299
  base64_image = base64.b64encode(cv2.imencode('.jpg', image_rgb)[1]).decode()
300
+ api_key = os.environ.get("OPENAI_API_KEY")
301
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
302
  payload = {
303
  "model": "gpt-4o-mini",
 
364
  gen_width, gen_height = resize_to_multiple_of_64(gen_width, gen_height)
365
 
366
  base64_image = base64.b64encode(cv2.imencode('.jpg', image_rgb)[1]).decode()
367
+ api_key = os.environ.get("OPENAI_API_KEY")
368
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
369
  payload = {
370
  "model": "gpt-4o-mini",