Mark-Lasfar
commited on
Commit
·
68e378c
1
Parent(s):
00269b2
endpoints.py generation.py
Browse files- utils/generation.py +3 -3
utils/generation.py
CHANGED
|
@@ -105,9 +105,9 @@ def select_model(query: str, input_type: str = "text", preferred_model: Optional
|
|
| 105 |
r"\bgenerate\s+image\b", r"\bcreate\s+image\b", r"\bimage\s+generation\b", r"\bصورة\s+توليد\b",
|
| 106 |
r"\bimage\s+edit\b", r"\bتحرير\s+صورة\b"
|
| 107 |
]
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
model = CLIP_LARGE_MODEL if preferred_model == "image_advanced" else CLIP_BASE_MODEL # ←
|
| 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:
|
|
|
|
| 105 |
r"\bgenerate\s+image\b", r"\bcreate\s+image\b", r"\bimage\s+generation\b", r"\bصورة\s+توليد\b",
|
| 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:
|