Update main.py

main.py

@@ -5,6 +5,7 @@ from transformers import pipeline
 import uvicorn
 from PIL import Image
 import io
+import os
 import numpy as np
 from projects.DL_CatDog.DL_CatDog import preprocess_image, read_image, model_DL_CatDog
 from projects.ML_StudentPerformance.ML_StudentPerformace import predict_student_performance, create_custom_data, form1