Lhumpal commited on
Commit
a39bfbf
·
verified ·
1 Parent(s): 0e7901e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -46,6 +46,8 @@ async def chat(request: ChatRequest):
46
  return {"response": response.choices[0].message.content}
47
 
48
  if request.model_choice == "google":
 
 
49
  messages = [
50
  {"role": "user", "parts": [{"text": "I have 10 dogs"}]},
51
  {"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},
 
46
  return {"response": response.choices[0].message.content}
47
 
48
  if request.model_choice == "google":
49
+ client = genai.Client(api_key=google_api_key)
50
+
51
  messages = [
52
  {"role": "user", "parts": [{"text": "I have 10 dogs"}]},
53
  {"role": "model", "parts": [{"text": "Great! Dogs are fun pets."}]},