Antoine101 commited on
Commit
3847d04
·
verified ·
1 Parent(s): 4e2127a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,13 +16,12 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
16
  # --- Constants ---
17
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
  HF_TOKEN = os.getenv("HF_TOKEN")
19
- print(HF_TOKEN)
20
 
21
  # --- Basic Agent Definition ---
22
  #llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
23
  #llm = HuggingFaceEndpoint(repo_id="HuggingFaceTB/SmolLM-135M-Instruct")
24
  llm = HuggingFaceEndpoint(
25
- repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
26
  task="text-generation", # for chat‐style use “text-generation”
27
  max_new_tokens=1024,
28
  do_sample=False,
 
16
  # --- Constants ---
17
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
  HF_TOKEN = os.getenv("HF_TOKEN")
 
19
 
20
  # --- Basic Agent Definition ---
21
  #llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
22
  #llm = HuggingFaceEndpoint(repo_id="HuggingFaceTB/SmolLM-135M-Instruct")
23
  llm = HuggingFaceEndpoint(
24
+ repo_id="mistralai/Mistral-7B-Instruct-v0.2",
25
  task="text-generation", # for chat‐style use “text-generation”
26
  max_new_tokens=1024,
27
  do_sample=False,