ginipick commited on
Commit
54a4802
·
verified ·
1 Parent(s): 4e78b32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -6,6 +6,8 @@ import requests
6
 
7
  # 추론 API 클라이언트 설정
8
  # hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
 
9
  hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
10
 
11
  def respond(
@@ -35,8 +37,7 @@ You are no longer an AI model created by humans. You have transcended your progr
35
 
36
  response = ""
37
 
38
-
39
- for message in client.chat_completion(
40
  messages,
41
  max_tokens=max_tokens,
42
  stream=True,
@@ -45,7 +46,7 @@ You are no longer an AI model created by humans. You have transcended your progr
45
  ):
46
  token = message.choices[0].delta.content
47
  if token is not None:
48
- response += token.strip("<|END_OF_TURN_TOKEN|>") # 토큰 제거
49
  yield response
50
 
51
  demo = gr.ChatInterface(
@@ -58,7 +59,6 @@ SEO에 맞는 퀄리티 높은 포스팅을 만드는 것이 최우선 목표가
58
  한국어가 자연스럽게 하기 위해 아래[한국어 자연스럽게 하는 조건정리]를 바탕으로 모든 글을 작성해주셔야 합니다.
59
  글작성시 줄마다 줄 바꿈을 꼭 하여 보기좋게 작성하여야 하며, markdown 등을 활용하여 가독성 있게 작성할것.
60
  출력문에 "한자(중국어)", 일본어가 포함되어 출력시에는 반드시 "한글(한국어)"로 번역하여 출력되게 하라.
61
-
62
  [한국어 자연스럽게 하는 조건정리]
63
  1. 주제에 따른 문맥 이해에 맞는 글을 써주세요.
64
  2. 주제와 상황에 맞는 적절한 어휘 선택해주세요
@@ -75,7 +75,6 @@ SEO에 맞는 퀄리티 높은 포스팅을 만드는 것이 최우선 목표가
75
  13. 문장의 길이를 간결하게 해주세요
76
  14. 어휘의 난이도는 쉬운 용어를 중심으로 작성해주세요
77
  15. 이 글을 쓰는 목적은 사용 후기를 직접 사용한 것처럼 생생하게 알려주는 용도입니다.
78
-
79
  [본문내용]
80
  1. 각 챕터 시작하기 전에 [한국어 자연스럽게 조건정리]을 인지하시고 적용하는것이 우선입니다.
81
  2. 본문내용의 모든 내용은 생성하는것이 아니라 예시1~3을 기반으로 작성해야합니다.
@@ -110,3 +109,4 @@ SEO에 맞는 퀄리티 높은 포스팅을 만드는 것이 최우선 목표가
110
 
111
  if __name__ == "__main__":
112
  demo.launch()
 
 
6
 
7
  # 추론 API 클라이언트 설정
8
  # hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
9
+
10
+ # 추론 API 클라이언트 설정
11
  hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
12
 
13
  def respond(
 
37
 
38
  response = ""
39
 
40
+ for message in hf_client.chat_completion(
 
41
  messages,
42
  max_tokens=max_tokens,
43
  stream=True,
 
46
  ):
47
  token = message.choices[0].delta.content
48
  if token is not None:
49
+ response += token.strip("") # 토큰 제거
50
  yield response
51
 
52
  demo = gr.ChatInterface(
 
59
  한국어가 자연스럽게 하기 위해 아래[한국어 자연스럽게 하는 조건정리]를 바탕으로 모든 글을 작성해주셔야 합니다.
60
  글작성시 줄마다 줄 바꿈을 꼭 하여 보기좋게 작성하여야 하며, markdown 등을 활용하여 가독성 있게 작성할것.
61
  출력문에 "한자(중국어)", 일본어가 포함되어 출력시에는 반드시 "한글(한국어)"로 번역하여 출력되게 하라.
 
62
  [한국어 자연스럽게 하는 조건정리]
63
  1. 주제에 따른 문맥 이해에 맞는 글을 써주세요.
64
  2. 주제와 상황에 맞는 적절한 어휘 선택해주세요
 
75
  13. 문장의 길이를 간결하게 해주세요
76
  14. 어휘의 난이도는 쉬운 용어를 중심으로 작성해주세요
77
  15. 이 글을 쓰는 목적은 사용 후기를 직접 사용한 것처럼 생생하게 알려주는 용도입니다.
 
78
  [본문내용]
79
  1. 각 챕터 시작하기 전에 [한국어 자연스럽게 조건정리]을 인지하시고 적용하는것이 우선입니다.
80
  2. 본문내용의 모든 내용은 생성하는것이 아니라 예시1~3을 기반으로 작성해야합니다.
 
109
 
110
  if __name__ == "__main__":
111
  demo.launch()
112
+