ishworrsubedii commited on
Commit
36f6588
·
1 Parent(s): 89b2e6e

fix: add model name while saving cto outputs

Browse files
src/components/product_page_nto_cto_gen.py CHANGED
@@ -209,7 +209,7 @@ class ProductPageImageGeneration:
209
  cto_image = requests.get(cto_response['output'])
210
 
211
  # Save clothing result
212
- filename = self.generate_filename(model_id, "cto", clothing_type)
213
  cto_url = self.upload_to_supabase(cto_image.content, filename)
214
 
215
  if cto_url:
 
209
  cto_image = requests.get(cto_response['output'])
210
 
211
  # Save clothing result
212
+ filename = self.generate_filename(base_id=model_id, model_name=model_name,try_on_type="cto", item_type=clothing_type)
213
  cto_url = self.upload_to_supabase(cto_image.content, filename)
214
 
215
  if cto_url: