JairoCesar commited on
Commit
da890ce
·
verified ·
1 Parent(s): 0a58fe0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,10 +8,11 @@ import re
8
  import os
9
 
10
  hf_token = os.getenv("HF_TOKEN")
 
11
 
12
  @st.cache_resource
13
  def get_inference_client():
14
- return InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
15
 
16
  def extract_and_clean_json(text):
17
  json_match = re.search(r'\{.*\}', text, re.DOTALL)
 
8
  import os
9
 
10
  hf_token = os.getenv("HF_TOKEN")
11
+ gemini_token=os.getenv("GEMINI_TOKEN")
12
 
13
  @st.cache_resource
14
  def get_inference_client():
15
+ return InferenceClient("gemini-2.0-flash")
16
 
17
  def extract_and_clean_json(text):
18
  json_match = re.search(r'\{.*\}', text, re.DOTALL)