mgokg commited on
Commit
d1b76e1
·
verified ·
1 Parent(s): 1739a6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,13 +47,13 @@ def selenium(message):
47
  time.sleep(3)
48
  element = wd.find_element(By.TAG_NAME, "body")
49
  #time.sleep(3)
50
- return element.text
51
  try:
52
  completion = client.chat.completions.create(
53
  model="llama3-8b-8192",
54
  messages=[
55
  {"role": "system", "content": "You are a helpful assistant."},
56
- {"role": "user", "content": f"return json object with keys name and email. \n name = {message}\n value for email can be found here: {response.text} \n return json object only, no additional text or comments \n"}
57
  ],
58
  )
59
  return completion.choices[0].message.content
 
47
  time.sleep(3)
48
  element = wd.find_element(By.TAG_NAME, "body")
49
  #time.sleep(3)
50
+ #return element.text
51
  try:
52
  completion = client.chat.completions.create(
53
  model="llama3-8b-8192",
54
  messages=[
55
  {"role": "system", "content": "You are a helpful assistant."},
56
+ {"role": "user", "content": f"return json object with keys name and email. \n name = {message}\n value for email can be found here: {element.text} \n return json object only, no additional text or comments \n"}
57
  ],
58
  )
59
  return completion.choices[0].message.content