Update handler.py

handler.py

@@ -27,7 +27,7 @@ def _ensure_llava():
 _ensure_llava()
 
 # ---- LLaVA parçaları (demo akışı) ----
-from llava.model.builder import load_pretrained_model, get_model_name_from_path
+from llava.model.builder import load_pretrained_model
 from llava.mm_utils import tokenizer_image_token, process_images
 from llava.constants import (
     IMAGE_TOKEN_INDEX,
@@ -38,6 +38,13 @@ from llava.constants import (
 from llava.conversation import conv_templates
 from llava.utils import disable_torch_init
 
+# Eksik fonksiyonu kendimiz tanımlıyoruz
+def get_model_name_from_path(model_path):
+    """Extract model name from path"""
+    if "/" in model_path:
+        return model_path.split("/")[-1]
+    return model_path
+
 # Varsayılanlar
 DEFAULT_CONV_MODE  = os.getenv("LLAVA_CONV_MODE", "llava_v2")
 MAX_NEW_TOKENS_DEF = int(os.getenv("MAX_NEW_TOKENS", "256"))
@@ -57,7 +64,7 @@ class EndpointHandler:
     def __init__(self, path: str = "") -> None:
         disable_torch_init()
 
-        # PULSE-7B HF’den/yerelden nereden yükleniyorsa yolu belirle
+        # PULSE-7B HF'den/yerelden nereden yükleniyorsa yolu belirle
         if os.getenv("HF_MODEL_LOCAL_DIR", "").strip():
             model_path = os.getenv("HF_MODEL_LOCAL_DIR").strip()
         elif os.getenv("HF_MODEL_ID", "").strip():
@@ -136,7 +143,7 @@ class EndpointHandler:
                 image_tensors = process_images([pil], self.image_processor, self.model.config)
                 image_tensors = image_tensors.to(self.model.device, dtype=self.model.dtype, non_blocking=True)
 
-        # 3) tokenize (image token’ı gömülü)
+        # 3) tokenize (image token'ı gömülü)
         input_ids = tokenizer_image_token(
             prompt, self.tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt"
         ).to(self.model.device, non_blocking=True)
@@ -159,4 +166,4 @@ class EndpointHandler:
             output_ids = self.model.generate(input_ids, images=image_tensors, **gen_kwargs)
 
         text = self.tokenizer.batch_decode(output_ids, skip_special_tokens=True)[0].strip()
-        return [{"generated_text": text}]
+        return [{"generated_text": text}]