AbuAlone09 commited on
Commit
c36bf37
·
verified ·
1 Parent(s): 118479c

Update backend_models.py

Browse files
Files changed (1) hide show
  1. backend_models.py +4 -2
backend_models.py CHANGED
@@ -4,8 +4,10 @@ import google.generativeai as genai
4
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
5
 
6
  def get_inference_client(model_id: str, provider: str = "auto"):
7
- model = genai.GenerativeModel('gemini-1.5-flash')
8
- return model
 
 
9
 
10
  def get_real_model_id(model_id: str) -> str:
11
  return "gemini-1.5-flash"
 
4
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
5
 
6
  def get_inference_client(model_id: str, provider: str = "auto"):
7
+ # Bỏ chữ 'models/' đi nếu nó đang bị gắn vào trước tên model
8
+ model_name = 'gemini-1.5-flash'
9
+ return genai.GenerativeModel(model_name)
10
+
11
 
12
  def get_real_model_id(model_id: str) -> str:
13
  return "gemini-1.5-flash"