Spaces:
Sleeping
Sleeping
Pavel commited on
Commit ·
c8a3432
1
Parent(s): 36c262d
gated update
Browse files- app/main.py +1 -1
app/main.py
CHANGED
|
@@ -5,7 +5,7 @@ import os
|
|
| 5 |
|
| 6 |
hf_token = os.getenv("token")
|
| 7 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
| 8 |
-
pipe_llama = pipeline("text2text-generation", model="meta-llama/Llama-2-70b-chat-hf",
|
| 9 |
|
| 10 |
app = FastAPI()
|
| 11 |
|
|
|
|
| 5 |
|
| 6 |
hf_token = os.getenv("token")
|
| 7 |
pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
|
| 8 |
+
pipe_llama = pipeline("text2text-generation", model="meta-llama/Llama-2-70b-chat-hf", token=hf_token)
|
| 9 |
|
| 10 |
app = FastAPI()
|
| 11 |
|