Spaces:
Sleeping
Sleeping
traopia
commited on
Commit
·
54ab71a
1
Parent(s):
78a4f9f
llm
Browse files- src/use_llm.py +0 -3
src/use_llm.py
CHANGED
|
@@ -31,9 +31,6 @@ def main_generate(prompt, model=DEFAULT_MODEL, system_prompt="You are a helpful
|
|
| 31 |
|
| 32 |
|
| 33 |
|
| 34 |
-
import os
|
| 35 |
-
from huggingface_hub import InferenceClient
|
| 36 |
-
|
| 37 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 38 |
MODEL_ID = "thenlper/gte-large" # embedding model
|
| 39 |
|
|
|
|
| 31 |
|
| 32 |
|
| 33 |
|
|
|
|
|
|
|
|
|
|
| 34 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 35 |
MODEL_ID = "thenlper/gte-large" # embedding model
|
| 36 |
|