Update src/app/response.py

src/app/response.py

@@ -50,7 +50,7 @@ def caption_image(image: PIL.Image.Image, max_new_tokens: int) -> str:
         logging.info("Caption generated successfully.")
 
         # Return the generated caption
-        return result[0][len(text) :].lstrip("\n")
+        return result[0][len(prompt) :].lstrip("\n")
 
     # Handle exceptions that may occur during caption generation
     except Exception as e: