Update main.py

main.py

@@ -11,17 +11,15 @@ 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
 
-# Set cache directory to /tmp/mycache
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/mycache'
-if not os.path.exists('/tmp/mycache'):
-    os.makedirs('/tmp/mycache')
-
-# Initialize the pipeline
-pipe = pipeline(
-    "image-classification", 
-    model="wambugu71/crop_leaf_diseases_vit", 
-    cache_dir='/tmp/mycache'
-)
+# Set the cache directory to a writable location
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/.cache'
+
+# Make sure the directory exists
+os.makedirs('/tmp/.cache', exist_ok=True)
+
+# Initialize the pipeline with the new cache directory
+pipe = pipeline("image-classification", model="wambugu71/crop_leaf_diseases_vit", cache_dir="/tmp/.cache")
+
 
 app = FastAPI()
 
@@ -52,7 +50,6 @@ 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", cache_dir="/tmp/mycache")
 @app.post("/api/classify")
 async def classify_image(file: UploadFile = File(...)):
     try: