userIdc2024 commited on
Commit
cf6d25f
·
verified ·
1 Parent(s): f275e0a

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -117,7 +117,7 @@ def _process_one(aspect_ratio: str, category: str, keyword: str, subkeyword: str
117
  image_urls = generate_images(prompt_text, aspect_ratio)
118
  image_bytes = fetch_image_bytes(image_urls)
119
  image_with_metadata = meta_data_helper_function(image_bytes)
120
- r2_url = upload_to_r2(image_bytes)
121
  return r2_url
122
 
123
  def get_images(category: str, keyword: str, subkeyword: str) -> List[str]:
 
117
  image_urls = generate_images(prompt_text, aspect_ratio)
118
  image_bytes = fetch_image_bytes(image_urls)
119
  image_with_metadata = meta_data_helper_function(image_bytes)
120
+ r2_url = upload_to_r2(image_with_metadata)
121
  return r2_url
122
 
123
  def get_images(category: str, keyword: str, subkeyword: str) -> List[str]: