Update main.py

main.py

@@ -34,7 +34,8 @@ plant_disease_dict = {
     "Corn": ['Corn___Cercospora_leaf_spot Gray_leaf_spot', 'Corn___Common_rust',
              'Corn___Northern_Leaf_Blight', 'Corn___healthy']
 }
-
+TMP_DIR = os.getenv("TMP_DIR", "/app/temp")
+os.makedirs(TMP_DIR, exist_ok=True)
 # Custom Self-Attention Layer
 @tf.keras.utils.register_keras_serializable()
 class SelfAttention(Layer):
@@ -100,7 +101,10 @@ async def predict_plant_disease(plant_name: str, file: UploadFile = File(...)):
         raise HTTPException(status_code=404, detail=f"Model file '{plant_name}_model.keras' not found")
 
     # Save uploaded file temporarily
-    temp_path = f"temp_image_{file.filename}"
+    
+
+    # Define the temp file path
+    temp_path = os.path.join(TMP_DIR, file.filename)
     with open(temp_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)