Spaces:
Paused
Paused
Update backend_models.py
Browse files- 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 |
-
|
| 8 |
-
|
|
|
|
|
|
|
| 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"
|