KalooAI commited on
Commit
0e3570b
·
verified ·
1 Parent(s): a4543a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -43,12 +43,13 @@ client = InferenceClient(HF_MODEL_ID, token=os.getenv("HF_TOKEN"))
43
  # return _llm
44
 
45
  llm = HuggingFaceEndpoint(
46
- repo_id="ibm-granite/granite-3.1-8b-instruct", # Using Granite for 2026 stability
 
47
  task="text-generation",
48
  max_new_tokens=512,
49
  temperature=0.1,
50
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
51
- wait_for_model=True
52
  )
53
 
54
 
 
43
  # return _llm
44
 
45
  llm = HuggingFaceEndpoint(
46
+ repo_id="meta-llama/Llama-3.1-8B-Instruct", # This is the "Live" 2026 model
47
+ # repo_id="ibm-granite/granite-3.1-8b-instruct", # Using Granite for 2026 stability
48
  task="text-generation",
49
  max_new_tokens=512,
50
  temperature=0.1,
51
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
52
+ wait_for_model=True
53
  )
54
 
55