Commit
·
2526eca
1
Parent(s):
2dda86f
Set fixed caption text for /colorize responses
Browse files- 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 |
-
#
|
| 715 |
-
|
| 716 |
-
|
| 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,
|