Update api_server.py

api_server.py

@@ -5,12 +5,13 @@ from PIL import Image
 import torchvision.transforms as transforms
 from pathlib import Path
 from ultralytics import YOLO
+import io
 
 # Disable tensorflow warnings
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
 from tensorflow import keras
-from flask import Flask, jsonify, request, render_template
+from flask import Flask, jsonify, request, render_template, send_file
 import torch
 
 load_type = 'local'