Update app.py

app.py

@@ -61,13 +61,11 @@ def get_image_description(image):
     torch.cuda.empty_cache()
     gc.collect()
 
-    descriptions = []
     prompt = "[INST] <image>\nDescribe the image in a sentence [/INST]"
 
     inputs = processor(prompt, image, return_tensors="pt").to("cuda:0")
     output = vision_model.generate(**inputs, max_new_tokens=100)
-    descriptions.append(processor.decode(output[0], skip_special_tokens=True))
-    return descriptions
+    return (processor.decode(output[0], skip_special_tokens=True))
 
 
 CSS = """
@@ -100,10 +98,13 @@ 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])[0])
-        # except:
-        #     descs.append("Could not generate image description due to some error")
+        try:
+            descs.append(img_doc_files[i]+"\n"+get_image_description(images[i]))
+        except:
+            descs.append("Could not generate image description due to some error")
+            print(img_doc_files)
+            print(images[i])
+            print()
         print(descs[-1])
         print()