Garvitj commited on
Commit
5903789
·
verified ·
1 Parent(s): f7c66ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,10 +5,10 @@ import io
5
  import mysql.connector
6
  import os
7
 
8
- HF_API_KEY = os.getenv("HF_API_KEY")
9
 
10
  # Hugging Face API Key (Replace with your actual key)
11
- client = InferenceClient(model="mistralai/Mistral-7B-v0.1", token=HF_API_KEY)
12
 
13
  def classify_task(user_input):
14
  """Classifies user task using Mistral."""
 
5
  import mysql.connector
6
  import os
7
 
8
+ HF_API_KEY = os.getenv("hf_key")
9
 
10
  # Hugging Face API Key (Replace with your actual key)
11
+ client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.3", token=HF_API_KEY)
12
 
13
  def classify_task(user_input):
14
  """Classifies user task using Mistral."""