Nick Starkov commited on
Commit
f232d53
·
1 Parent(s): 7ec4963
Files changed (2) hide show
  1. app.py +4 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,8 +1,12 @@
 
1
  from fastapi import FastAPI
2
  from datasets import load_dataset
3
  import random
4
  import json
5
 
 
 
 
6
  app = FastAPI()
7
 
8
  # Load the dataset from Hugging Face
 
1
+ import os
2
  from fastapi import FastAPI
3
  from datasets import load_dataset
4
  import random
5
  import json
6
 
7
+ # Set Hugging Face cache directory programmatically as a fallback
8
+ os.environ["HF_HOME"] = "/app/cache"
9
+
10
  app = FastAPI()
11
 
12
  # Load the dataset from Hugging Face
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
- flask==2.3.3
 
2
  datasets==2.21.0
 
1
+ fastapi==0.115.0
2
+ uvicorn==0.32.0
3
  datasets==2.21.0