Update fastapi_app.py

fastapi_app.py

@@ -3,11 +3,11 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import JSONResponse, RedirectResponse
 from tempfile import NamedTemporaryFile
 import whisper
-import torch
 from typing import List
 
 # Set a writable cache directory for Whisper models
-os.environ["XDG_CACHE_HOME"] = "/app/.cache"  # Use a writable directory
+os.makedirs("/app/whisper_cache", exist_ok=True)  # Create a writable directory
+os.environ["XDG_CACHE_HOME"] = "/app/whisper_cache"  # Use the new writable directory
 
 # Checking if NVIDIA GPU is available
 DEVICE = "cpu"