Spaces:
Runtime error
Runtime error
arxivgpt kim
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -7,20 +7,22 @@ import numpy as np
|
|
7 |
# OpenAI API 키 설정
|
8 |
openai.api_key = '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'
|
9 |
|
|
|
10 |
# 문장 임베딩 모델 로드
|
11 |
model = SentenceTransformer('all-MiniLM-L6-v2')
|
12 |
|
13 |
def summarize_and_find_similar_sentence(input_text):
|
14 |
# GPT-3.5-turbo를 사용하여 입력 텍스트 요약
|
15 |
-
response = openai.
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
|
|
21 |
)
|
22 |
|
23 |
-
summary = response.choices[0].
|
24 |
|
25 |
# 입력 텍스트와 요약된 내용을 문장 임베딩으로 변환
|
26 |
input_embedding = model.encode(input_text, convert_to_tensor=True)
|
|
|
7 |
# OpenAI API 키 설정
|
8 |
openai.api_key = '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'
|
9 |
|
10 |
+
|
11 |
# 문장 임베딩 모델 로드
|
12 |
model = SentenceTransformer('all-MiniLM-L6-v2')
|
13 |
|
14 |
def summarize_and_find_similar_sentence(input_text):
|
15 |
# GPT-3.5-turbo를 사용하여 입력 텍스트 요약
|
16 |
+
response = openai.Completion.create(
|
17 |
+
engine="text-davinci-003", # 모델을 지정합니다.
|
18 |
+
prompt=f"Summarize this: {input_text}", # 요약을 위한 프롬프트를 설정합니다.
|
19 |
+
max_tokens=100, # 최대 토큰 수를 지정합니다.
|
20 |
+
n=1, # 생성할 완성 횟수를 지정합니다.
|
21 |
+
stop=None, # 완성을 멈출 문자 또는 문자열 목록을 지정합니다.
|
22 |
+
temperature=0.5, # 창의성 수준을 설정합니다.
|
23 |
)
|
24 |
|
25 |
+
summary = response.choices[0].text.strip()
|
26 |
|
27 |
# 입력 텍스트와 요약된 내용을 문장 임베딩으로 변환
|
28 |
input_embedding = model.encode(input_text, convert_to_tensor=True)
|