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

endpoints.py generation.py

Browse files
Files changed (1) hide show
  1. 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
- 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 # ← هنا زودت 4 مسافات
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: