Update flask/app.py

flask/app.py

@@ -8,6 +8,7 @@ import cv2
 import numpy as np
 import gradio as gr
 from time import gmtime, strftime
+from pydantic import BaseModel
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
 from typing import Dict