seawolf2357 commited on
Commit
746792d
โ€ข
1 Parent(s): bad7ad6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -45,8 +45,10 @@ async def generate_response(user_input):
45
 
46
  # ๋™๊ธฐ ํ•จ์ˆ˜๋ฅผ ๋น„๋™๊ธฐ๋กœ ์ฒ˜๋ฆฌํ•˜๊ธฐ ์œ„ํ•œ ๋ž˜ํผ ์‚ฌ์šฉ
47
  loop = asyncio.get_event_loop()
48
- response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(messages, max_tokens=4000, stream=False, temperature=0.7, top_p=0.95))
49
- return response['choices'][0]['text'] # 'text' ์†์„ฑ ์‚ฌ์šฉ
 
 
50
 
51
  # ๋””์Šค์ฝ”๋“œ ๋ด‡ ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ ๋ฐ ์‹คํ–‰
52
  discord_client = MyClient(intents=intents)
 
45
 
46
  # ๋™๊ธฐ ํ•จ์ˆ˜๋ฅผ ๋น„๋™๊ธฐ๋กœ ์ฒ˜๋ฆฌํ•˜๊ธฐ ์œ„ํ•œ ๋ž˜ํผ ์‚ฌ์šฉ
47
  loop = asyncio.get_event_loop()
48
+ response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(messages, max_tokens=1000, stream=False, temperature=0.7, top_p=0.95))
49
+ # dataclass ์†์„ฑ ์ ‘๊ทผ์œผ๋กœ ์ˆ˜์ •
50
+ return response.choices[0].text.strip()
51
+
52
 
53
  # ๋””์Šค์ฝ”๋“œ ๋ด‡ ์ธ์Šคํ„ด์Šค ์ƒ์„ฑ ๋ฐ ์‹คํ–‰
54
  discord_client = MyClient(intents=intents)