Update app.py

app.py

@@ -18,13 +18,6 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 extract_model = YOLO("best.pt").to(device)
 detect_model  = YOLO("yolov8n.pt").to(device)
 
-# Load LLaVA-HF LLM and processor
-processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.5-7b-hf")
-llava = LlavaNextForConditionalGeneration.from_pretrained(
-    "llava-hf/llava-v1.5-7b-hf",
-    torch_dtype=torch.float16 if device == "cuda" else torch.float32,
-    low_cpu_mem_usage=True
-).to(device)
 
 def caption_image_with_llava(image_path):
     image = Image.open(image_path).convert("RGB")