Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -34,7 +34,7 @@ import google.generativeai as genai
|
|
| 34 |
|
| 35 |
GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY')
|
| 36 |
genai.configure(api_key=GOOGLE_API_KEY)
|
| 37 |
-
model = genai.GenerativeModel('gemini-
|
| 38 |
|
| 39 |
# API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
|
| 40 |
def query(payload):
|
|
|
|
| 34 |
|
| 35 |
GOOGLE_API_KEY = os.environ.get('GOOGLE_API_KEY')
|
| 36 |
genai.configure(api_key=GOOGLE_API_KEY)
|
| 37 |
+
model = genai.GenerativeModel('gemini-2.0-flash')
|
| 38 |
|
| 39 |
# API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
|
| 40 |
def query(payload):
|