Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,12 +8,11 @@ from langsmith import Client
|
|
| 8 |
from typing import Dict
|
| 9 |
|
| 10 |
# Initialize LangSmith Client for monitoring
|
| 11 |
-
client = Client(api_key="
|
| 12 |
|
| 13 |
-
# Load Hugging Face LLM model
|
| 14 |
def load_huggingface_llm():
|
| 15 |
-
return pipeline("text-generation", model="
|
| 16 |
-
token=os.environ.get("hf_token")) # Ensure you set your token in the environment
|
| 17 |
|
| 18 |
llm = load_huggingface_llm()
|
| 19 |
|
|
|
|
| 8 |
from typing import Dict
|
| 9 |
|
| 10 |
# Initialize LangSmith Client for monitoring
|
| 11 |
+
client = Client(api_key="lsv2_pt_c29f0a7a770c4b729fb4427262d2272_064d0193b4")
|
| 12 |
|
| 13 |
+
# Load Hugging Face LLM model
|
| 14 |
def load_huggingface_llm():
|
| 15 |
+
return pipeline("text-generation", model="distilgpt2") # Using distilgpt2
|
|
|
|
| 16 |
|
| 17 |
llm = load_huggingface_llm()
|
| 18 |
|