Sujithanumala commited on
Commit
e804b63
·
verified ·
1 Parent(s): 671465d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -12,11 +12,12 @@ genai.configure(api_key="AIzaSyD2o8vjePJb6z8vT_PVe82lVWMD3_cBL0g")
12
  def predict(message :str ,history : Optional[List],topic : Optional[str] = "OIC") -> str:
13
  model = genai.GenerativeModel("gemini-pro")
14
  his = []
15
- for i,j in history:
16
- his.extend([
17
- {"role": "user", "parts": i},
18
- {"role": "model", "parts": j},
19
- ])
 
20
  chat = model.start_chat(
21
  history=his
22
  )
 
12
  def predict(message :str ,history : Optional[List],topic : Optional[str] = "OIC") -> str:
13
  model = genai.GenerativeModel("gemini-pro")
14
  his = []
15
+ if history:
16
+ for i,j in history:
17
+ his.extend([
18
+ {"role": "user", "parts": i},
19
+ {"role": "model", "parts": j},
20
+ ])
21
  chat = model.start_chat(
22
  history=his
23
  )