Akshayram1 commited on
Commit
95ef8ac
·
verified ·
1 Parent(s): 00fdb55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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="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")
12
 
13
- # Load Hugging Face LLM model with your Hugging Face token
14
  def load_huggingface_llm():
15
- return pipeline("text-generation", model="mistral/Mistral-7B-Instruct-v0.2",
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