Update main.py

main.py

@@ -10,6 +10,8 @@ from projects.DL_CatDog.DL_CatDog import preprocess_image, read_image, model_DL_
 from projects.ML_StudentPerformance.ML_StudentPerformace import predict_student_performance, create_custom_data, form1
 from projects.ML_DiabetesPrediction.ML_DiabetesPrediction import model_ML_DiabetesPrediction, form2
 
+os.environ['HF_HOME'] = './mycache'
+
 app = FastAPI()
 
 # Add CORS middleware
@@ -21,6 +23,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+
 # Health check route
 @app.get("/api/working")
 def home():
@@ -39,7 +42,7 @@ async def predict_DL_CatDog(file: UploadFile = File(...)):
         return JSONResponse(content={"ok": -1, "message": f"Something went wrong! {str(e)}"}, status_code=500)
 
 # Classification route for DL_PlantDisease
-pipe = pipeline("image-classification", model="wambugu71/crop_leaf_diseases_vit")
+pipe = pipeline("image-classification", model="wambugu71/crop_leaf_diseases_vit", cache_dir="./mycache")
 @app.post("/api/classify")
 async def classify_image(file: UploadFile = File(...)):
     try: