Mark-Lasfar commited on
Commit
d6ba029
·
1 Parent(s): 68e378c

endpoints.py generation.py

Browse files
Files changed (1) hide show
  1. 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
- 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]}...")
 
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]}...")