kushagrasharma-13 commited on
Commit
4c6babf
·
1 Parent(s): a90831f

Add application file

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -13,6 +13,8 @@ import os
13
  import json
14
  from ctransformers import AutoModelForCausalLM
15
 
 
 
16
  app = FastAPI()
17
  templates = Jinja2Templates(directory="templates")
18
  app.mount("/static", StaticFiles(directory="static"), name="static")
 
13
  import json
14
  from ctransformers import AutoModelForCausalLM
15
 
16
+ os.environ["TRANSFORMERS_FORCE_CPU"] = "true"
17
+
18
  app = FastAPI()
19
  templates = Jinja2Templates(directory="templates")
20
  app.mount("/static", StaticFiles(directory="static"), name="static")
requirements.txt CHANGED
@@ -4,4 +4,5 @@ qdrant-client
4
  sentence-transformers
5
  uvicorn
6
  fastapi
7
- ctransformers
 
 
4
  sentence-transformers
5
  uvicorn
6
  fastapi
7
+ ctransformers
8
+ transformers