VitaliiPuzhenko commited on
Commit
cee5832
·
verified ·
1 Parent(s): a359e7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -2,12 +2,17 @@ import os
2
  from fastapi import FastAPI, Request
3
  from transformers import pipeline
4
 
 
 
 
 
 
5
  app = FastAPI()
6
 
7
  pipe = pipeline(
8
  "text-generation",
9
  model="google/gemma-3-1b-it",
10
- token=os.environ["HF_TOKEN"], # токен передається напряму
11
  device="cpu",
12
  max_new_tokens=256,
13
  temperature=0.7
 
2
  from fastapi import FastAPI, Request
3
  from transformers import pipeline
4
 
5
+ # 🔐 Повний контроль над кешем і токеном
6
+ os.environ["HF_TOKEN"] = os.environ["HF_TOKEN"]
7
+ os.environ["HF_HOME"] = "/data/.hf"
8
+ os.environ["TRANSFORMERS_CACHE"] = "/data/transformers"
9
+
10
  app = FastAPI()
11
 
12
  pipe = pipeline(
13
  "text-generation",
14
  model="google/gemma-3-1b-it",
15
+ token=os.environ["HF_TOKEN"], # 🔑 без login()
16
  device="cpu",
17
  max_new_tokens=256,
18
  temperature=0.7