AksharaSachin commited on
Commit
fe07d7a
·
verified ·
1 Parent(s): 2434d18

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -47,7 +47,7 @@ def build_agent():
47
  # huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
48
  # )
49
 
50
- llm = HuggingFaceHub(repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", task="text-generation",
51
  model_kwargs={
52
  "temperature": 0.7, # Controls randomness (0.0 = deterministic)
53
  "max_new_tokens": 100 # Optional: max output length
 
47
  # huggingfacehub_api_token=os.getenv("HUGGINGFACEHUB_API_TOKEN")
48
  # )
49
 
50
+ llm = HuggingFaceHub(repo_id="openai-community/gpt2-medium", task="text-generation",
51
  model_kwargs={
52
  "temperature": 0.7, # Controls randomness (0.0 = deterministic)
53
  "max_new_tokens": 100 # Optional: max output length