Update api.py

api.py

@@ -9,6 +9,7 @@ import google.generativeai as genai
 from pydantic import BaseModel
 from ultralytics import YOLO
 import matplotlib.pyplot as plt
+from io import BytesIO
 
 from langchain.document_loaders import TextLoader  # Or a custom loader for .docs
 from langchain.text_splitter import RecursiveCharacterTextSplitter