Update handler.py

handler.py

@@ -8,7 +8,7 @@ from serpapi import GoogleSearch
 from keybert import KeyBERT
 from typing import Dict, Any, List
 import base64
-
+import torch
 model_id = "vikhyatk/moondream2"
 revision = "2024-08-26"
 model = AutoModelForCausalLM.from_pretrained(
@@ -53,6 +53,7 @@ class ProductSearcher:
 
         enc_image = model.encode_image(image)
         self.description = model.answer_question(enc_image, prompt, tokenizer)
+        del enc_image
 
     def extract_keyphrases(self):
         self.keyphrases = self.kw_model.extract_keywords(self.description)