themeetjani commited on
Commit
5b7ed98
·
verified ·
1 Parent(s): 860d47f

Update pages/Our_model.py

Browse files
Files changed (1) hide show
  1. pages/Our_model.py +2 -1
pages/Our_model.py CHANGED
@@ -3,6 +3,7 @@ from openai import OpenAI
3
  import streamlit as st
4
  from streamlit import session_state
5
  import numpy as np
 
6
  import os
7
  client = OpenAI()
8
  openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -33,7 +34,7 @@ def gpt_scorer(m,s):
33
  frequency_penalty=0,
34
  presence_penalty=0
35
  )
36
- return response.choices[0].message.content
37
  def embeddings_cosine(s1,s2,model):
38
  embeddings1 = model.encode(s1)
39
  embeddings2 = model.encode(s2)
 
3
  import streamlit as st
4
  from streamlit import session_state
5
  import numpy as np
6
+ import json
7
  import os
8
  client = OpenAI()
9
  openai.api_key = os.getenv("OPENAI_API_KEY")
 
34
  frequency_penalty=0,
35
  presence_penalty=0
36
  )
37
+ return json.loads(response.choices[0].message.content)
38
  def embeddings_cosine(s1,s2,model):
39
  embeddings1 = model.encode(s1)
40
  embeddings2 = model.encode(s2)