phamluan commited on
Commit
242185f
·
1 Parent(s): d1ad02a
Files changed (1) hide show
  1. app/app.py +4 -3
app/app.py CHANGED
@@ -1,10 +1,12 @@
 
 
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
  from transformers import pipeline
4
 
5
  app = FastAPI()
6
 
7
- # Pipeline với model của bạn
8
  model_id = "phamluan/ai-auto-excute-script"
9
  classifier = pipeline("text-classification", model=model_id, tokenizer=model_id)
10
 
@@ -14,5 +16,4 @@ class InputText(BaseModel):
14
  @app.post("/predict")
15
  def predict(data: InputText):
16
  result = classifier(data.text)
17
- return {"result": result}
18
-
 
1
+ import os
2
+ os.environ["HF_HOME"] = "/tmp" # ✅ Tránh lỗi ghi vào /.cache
3
+
4
  from fastapi import FastAPI
5
  from pydantic import BaseModel
6
  from transformers import pipeline
7
 
8
  app = FastAPI()
9
 
 
10
  model_id = "phamluan/ai-auto-excute-script"
11
  classifier = pipeline("text-classification", model=model_id, tokenizer=model_id)
12
 
 
16
  @app.post("/predict")
17
  def predict(data: InputText):
18
  result = classifier(data.text)
19
+ return {"result": result}