Spaces:
Runtime error
Runtime error
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):
|