Initial commit from Gitpod

app.py

@@ -1,29 +1,27 @@
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-from transformers import pipeline
-from fastapi.responses import StreamingResponse
-import matplotlib.pyplot as plt
-import requests
-import datetime
-from io import BytesIO
-import logging
 import os
 
-# Set environment variables for Hugging Face cache
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
-os.environ["HF_HOME"] = "/tmp/hf_cache"
-import os
-
-# Set temp cache dirs
+# Redirect default cache paths to writable tmp
 os.environ["MPLCONFIGDIR"] = "/tmp/mplconfig"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
 os.environ["HF_HOME"] = "/tmp/huggingface"
 
-# Ensure the directories exist
+# Create the directories if they don't exist
 os.makedirs("/tmp/mplconfig", exist_ok=True)
 os.makedirs("/tmp/transformers", exist_ok=True)
 os.makedirs("/tmp/huggingface", exist_ok=True)
 
+
+
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from transformers import pipeline
+from fastapi.responses import StreamingResponse
+import matplotlib.pyplot as plt
+import requests
+import datetime
+from io import BytesIO
+import logging
+
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)