Update app.py

app.py

@@ -10,7 +10,7 @@ app = FastAPI()
 async def generate_text():
     try:
         # Get the cache directory from the environment variable
-        cache_dir = os.getenv("TRANSFORMERS_CACHE", "/code/cache")
+        cache_dir = os.getenv("HF_HOME")
 
         # Specify the directory for model download within the Docker container
         model_dir = os.path.join(cache_dir, "TheBloke/Mistral-7B-v0.1-GGUF")