Spaces:
Runtime error
Runtime error
fix: cache folder
Browse files
main.py
CHANGED
@@ -3,6 +3,9 @@ import uvicorn
|
|
3 |
from fastapi import FastAPI
|
4 |
from pydantic import BaseModel
|
5 |
from transformers import pipeline
|
|
|
|
|
|
|
6 |
|
7 |
logging.basicConfig(
|
8 |
format='%(asctime)s.%(msecs)03d %(levelname)-8s %(message)s',
|
|
|
3 |
from fastapi import FastAPI
|
4 |
from pydantic import BaseModel
|
5 |
from transformers import pipeline
|
6 |
+
import os
|
7 |
+
|
8 |
+
os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
|
9 |
|
10 |
logging.basicConfig(
|
11 |
format='%(asctime)s.%(msecs)03d %(levelname)-8s %(message)s',
|