Update app.py

app.py

@@ -1,11 +1,18 @@
+import os
 from fastapi import FastAPI
 from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast
 import torch
 
+# Set Hugging Face cache directory to a writable path
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
+
 app = FastAPI()
 
-model = DistilBertForSequenceClassification.from_pretrained("ADKU/ResearchGPT_model")
-tokenizer = DistilBertTokenizerFast.from_pretrained("ADKU/ResearchGPT_model")
+model_name = "ADKU/ResearchGPT_model"  # Replace with your actual Hugging Face model ID
+
+# Load model and tokenizer
+model = DistilBertForSequenceClassification.from_pretrained(model_name, cache_dir="/tmp/huggingface")
+tokenizer = DistilBertTokenizerFast.from_pretrained(model_name, cache_dir="/tmp/huggingface")
 
 @app.post("/predict/")
 async def predict(text: str):