LogicGoInfotechSpaces commited on
Commit
2526eca
·
1 Parent(s): 2dda86f

Set fixed caption text for /colorize responses

Browse files
Files changed (1) hide show
  1. app/main.py +4 -7
app/main.py CHANGED
@@ -710,13 +710,10 @@ async def colorize(
710
  base_url = "https://logicgoinfotechspaces-text-guided-image-colorization.hf.space"
711
 
712
  result_id_clean = result_id.replace(".png", "")
713
-
714
- # Generate caption based on model used
715
- if model_type == "cco":
716
- caption = f"Colorized using CCO {cco_model} model with vibrant natural colors, high quality"
717
- else:
718
- caption = "Colorized using GAN model with vibrant natural colors, high quality"
719
-
720
  response_data = {
721
  "success": True,
722
  "result_id": result_id_clean,
 
710
  base_url = "https://logicgoinfotechspaces-text-guided-image-colorization.hf.space"
711
 
712
  result_id_clean = result_id.replace(".png", "")
713
+
714
+ # Use a consistent caption text for all models
715
+ caption = "colorize this image with vibrant natural colors, high quality"
716
+
 
 
 
717
  response_data = {
718
  "success": True,
719
  "result_id": result_id_clean,