Update main.py

main.py

@@ -22,7 +22,7 @@ from tensorflow.keras.models import Model
 from tensorflow.keras.preprocessing.image import img_to_array
 from sklearn.metrics.pairwise import cosine_similarity
 
-BASE_DIR = "saved_data"
+BASE_DIR = "/tmp/data"
 app = FastAPI()
 mobilenet = MobileNetV2(weights="imagenet", include_top=False, pooling='avg')
 
@@ -114,10 +114,6 @@ def load_image(source, frame_time=1):
             os.remove(temp_video_path)
             return frame
 
-app = FastAPI()
-
-BASE_DIR = "/tmp/data"
-
 @app.post("/save")
 async def save(image_data: RequestModel):
     os.makedirs(BASE_DIR, exist_ok=True)