GaborToth2 commited on
Commit
8ffd026
·
1 Parent(s): 3bad752
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import cohere
6
  HF_API_KEY = os.getenv("HF_API_KEY")
7
  COHERE_API_KEY = os.getenv("COHERE_API_KEY") # Get Cohere API key
8
 
9
- models = ["HuggingFaceH4/zephyr-7b-beta", "microsoft/Phi-4-mini-instruct", "meta-llama/Llama-3.2-3B-Instruct", "meta-llama/Llama-3.1-8B-Instruct"]
10
- client_hf = InferenceClient(model=models[3], token=HF_API_KEY) # HF Client
11
  client_cohere = cohere.Client(COHERE_API_KEY) # Cohere Client
12
 
13
  def respond(
 
6
  HF_API_KEY = os.getenv("HF_API_KEY")
7
  COHERE_API_KEY = os.getenv("COHERE_API_KEY") # Get Cohere API key
8
 
9
+ models = ["HuggingFaceH4/zephyr-7b-beta", "meta-llama/Llama-3.2-3B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"]
10
+ client_hf = InferenceClient(model=models[2], token=HF_API_KEY) # HF Client
11
  client_cohere = cohere.Client(COHERE_API_KEY) # Cohere Client
12
 
13
  def respond(