KalooAI commited on
Commit
e40affa
·
verified ·
1 Parent(s): d82fd48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -63,11 +63,11 @@ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
63
  # 1. Use a model that has high availability in the 2026 Provider network
64
  # Llama-3.1-8B-Instruct is currently a 'gold standard' for this router
65
  # repo_id = "meta-llama/Llama-3.1-8B-Instruct"
66
- # repo_id = "Qwen/Qwen2.5-7B-Instruct"
67
  # repo_id = "google/gemma-2-2b-it"
68
  # repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
69
  # repo_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
70
- repo_id = "microsoft/Phi-3.5-mini-instruct"
71
 
72
 
73
  llm = HuggingFaceEndpoint(
@@ -75,6 +75,7 @@ llm = HuggingFaceEndpoint(
75
  task="text-generation",
76
  max_new_tokens=512,
77
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
 
78
  model_kwargs={"wait_for_model": True}
79
  )
80
 
 
63
  # 1. Use a model that has high availability in the 2026 Provider network
64
  # Llama-3.1-8B-Instruct is currently a 'gold standard' for this router
65
  # repo_id = "meta-llama/Llama-3.1-8B-Instruct"
66
+ repo_id = "Qwen/Qwen2.5-7B-Instruct"
67
  # repo_id = "google/gemma-2-2b-it"
68
  # repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
69
  # repo_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
70
+ # repo_id = "microsoft/Phi-3.5-mini-instruct"
71
 
72
 
73
  llm = HuggingFaceEndpoint(
 
75
  task="text-generation",
76
  max_new_tokens=512,
77
  huggingfacehub_api_token=os.getenv("HF_TOKEN"),
78
+ provider = 'together';
79
  model_kwargs={"wait_for_model": True}
80
  )
81