Update app.py

app.py

@@ -62,11 +62,11 @@ def get_image_description(image):
     gc.collect()
 
     prompt = "[INST] <image>\nDescribe the image in a sentence [/INST]"
-    n = len(prompt)
+    # n = len(prompt)
 
     inputs = processor(prompt, image, return_tensors="pt").to("cuda:0")
     output = vision_model.generate(**inputs, max_new_tokens=100)
-    return (processor.decode(output[0][n:], skip_special_tokens=True))
+    return (processor.decode(output[0], skip_special_tokens=True))
 
 
 CSS = """