Mark-Lasfar
commited on
Commit
·
d6ba029
1
Parent(s):
68e378c
endpoints.py generation.py
Browse files- utils/generation.py +4 -4
utils/generation.py
CHANGED
|
@@ -106,10 +106,10 @@ def select_model(query: str, input_type: str = "text", preferred_model: Optional
|
|
| 106 |
r"\bimage\s+edit\b", r"\bتحرير\s+صورة\b"
|
| 107 |
]
|
| 108 |
for pattern in image_patterns:
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 113 |
for pattern in image_gen_patterns:
|
| 114 |
if re.search(pattern, query_lower, re.IGNORECASE) or input_type == "image_gen":
|
| 115 |
logger.info(f"Selected {IMAGE_GEN_MODEL} with endpoint {FALLBACK_API_ENDPOINT} for image generation query: {query[:50]}...")
|
|
|
|
| 106 |
r"\bimage\s+edit\b", r"\bتحرير\s+صورة\b"
|
| 107 |
]
|
| 108 |
for pattern in image_patterns:
|
| 109 |
+
if re.search(pattern, query_lower, re.IGNORECASE):
|
| 110 |
+
model = CLIP_LARGE_MODEL if preferred_model == "image_advanced" else CLIP_BASE_MODEL # ← اضغط Tab هنا
|
| 111 |
+
logger.info(f"Selected {model} with endpoint {IMAGE_INFERENCE_API} for image-related query: {query[:50]}...")
|
| 112 |
+
return model, f"{IMAGE_INFERENCE_API}/{model}"
|
| 113 |
for pattern in image_gen_patterns:
|
| 114 |
if re.search(pattern, query_lower, re.IGNORECASE) or input_type == "image_gen":
|
| 115 |
logger.info(f"Selected {IMAGE_GEN_MODEL} with endpoint {FALLBACK_API_ENDPOINT} for image generation query: {query[:50]}...")
|