VitaliiPuzhenko commited on
Commit
ed0a7a7
·
verified ·
1 Parent(s): 8c6d1a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -1,18 +1,13 @@
1
- import os
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
 
 
1
  from fastapi import FastAPI, Request
2
  from transformers import pipeline
3
+ import os
 
 
 
 
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