arxivgpt kim commited on
Commit
6aa16b6
·
verified ·
1 Parent(s): 79e99c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
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.ChatCompletion.create(
16
- model="gpt-3.5-turbo",
17
- messages=[
18
- {"role": "system", "content": "You are a helpful assistant."},
19
- {"role": "user", "content": f"Summarize this: {input_text}"}
20
- ]
 
21
  )
22
 
23
- summary = response.choices[0].message.content
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)