Update app.py

app.py

@@ -55,13 +55,13 @@ if torch.cuda.is_available():
         load_in_4bit=True,
     )
 
+prompt = "[INST] <image>\nDescribe the image in a sentence [/INST]"
 
 @spaces.GPU()
 def get_image_description(image):
     torch.cuda.empty_cache()
     gc.collect()
 
-    prompt = "[INST] <image>\nDescribe the image in a sentence [/INST]"
     # n = len(prompt)
 
     inputs = processor(prompt, image, return_tensors="pt").to("cuda:0")
@@ -100,7 +100,7 @@ def get_vectordb(text, images, img_doc_files):
     descs = []
     for i in range(len(images)):
         try:
-            descs.append(img_doc_files[i]+"\n"+get_image_description(images[i]))
+            descs.append(img_doc_files[i]+"\n"+get_image_description(images[i]).replace(prompt, ""))
         except:
             descs.append("Could not generate image description due to some error")
         print(descs[-1])