Update main.py

main.py

@@ -12,10 +12,10 @@ from projects.ML_StudentPerformance.ML_StudentPerformace import predict_student_
 from projects.ML_DiabetesPrediction.ML_DiabetesPrediction import model_ML_DiabetesPrediction, form2
 
 # Set the cache directory to a writable location
-os.environ['TRANSFORMERS_CACHE'] = '/mnt/data/.cache'
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/.cache'
 
 # Make sure the directory exists
-os.makedirs('/mnt/data/.cache', exist_ok=True)
+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")