Spaces:
Sleeping
Sleeping
Update pages/Our_model.py
Browse files- 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)
|