seawolf2357 commited on
Commit
8af719d
Β·
verified Β·
1 Parent(s): d14b0d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -18,6 +18,14 @@ translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
18
  # κ³ μ •λœ λ„€κ±°ν‹°λΈŒ ν”„λ‘¬ν”„νŠΈ
19
  negative_prompt = "blur, low quality, bad composition, ugly, disfigured, weird colors, low quality, jpeg artifacts, lowres, grainy, deformed structures, blurry, opaque, low contrast, distorted details, details are low"
20
 
 
 
 
 
 
 
 
 
21
  # 이미지 생성 및 응닡 처리 ν•¨μˆ˜
22
  def generate_image(prompt, negative_prompt):
23
  headers = {
@@ -39,8 +47,6 @@ def generate_image(prompt, negative_prompt):
39
  logging.error(f"API μš”μ²­μ— μ‹€νŒ¨ν–ˆμŠ΅λ‹ˆλ‹€: {response.text}")
40
  return None
41
 
42
-
43
-
44
  class MyClient(discord.Client):
45
  async def on_ready(self):
46
  logging.info(f'{self.user}둜 λ‘œκ·ΈμΈλ˜μ—ˆμŠ΅λ‹ˆλ‹€!')
 
18
  # κ³ μ •λœ λ„€κ±°ν‹°λΈŒ ν”„λ‘¬ν”„νŠΈ
19
  negative_prompt = "blur, low quality, bad composition, ugly, disfigured, weird colors, low quality, jpeg artifacts, lowres, grainy, deformed structures, blurry, opaque, low contrast, distorted details, details are low"
20
 
21
+ # ν”„λ‘¬ν”„νŠΈ λ²ˆμ—­ ν•¨μˆ˜
22
+ def translate_prompt(prompt):
23
+ logging.debug(f'ν”„λ‘¬ν”„νŠΈ λ²ˆμ—­ 쀑: {prompt}')
24
+ translation = translator(prompt, max_length=512)
25
+ translated_text = translation[0]['translation_text']
26
+ logging.debug(f'λ²ˆμ—­λœ ν…μŠ€νŠΈ: {translated_text}')
27
+ return translated_text
28
+
29
  # 이미지 생성 및 응닡 처리 ν•¨μˆ˜
30
  def generate_image(prompt, negative_prompt):
31
  headers = {
 
47
  logging.error(f"API μš”μ²­μ— μ‹€νŒ¨ν–ˆμŠ΅λ‹ˆλ‹€: {response.text}")
48
  return None
49
 
 
 
50
  class MyClient(discord.Client):
51
  async def on_ready(self):
52
  logging.info(f'{self.user}둜 λ‘œκ·ΈμΈλ˜μ—ˆμŠ΅λ‹ˆλ‹€!')