Update app.py

app.py

@@ -8,7 +8,6 @@ import torch
 from diffusers import DiffusionPipeline
 import io
 from PIL import Image
-import time
 from dotenv import load_dotenv
 
 # .env 파일에서 환경 변수 로드
@@ -24,17 +23,20 @@ intents.message_content = True
 # 번역 파이프라인 설정
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
-# 환경 변수에서 지정된 채널 ID 가져오기, 기본값 사용
-TARGET_CHANNEL_ID = os.getenv('DISCORD_CHANNEL_ID')
+# 환경 변수에서 지정된 채널 ID 가져오기
+TARGET_CHANNEL_ID = int(os.getenv('DISCORD_CHANNEL_ID'))  # 채널 ID는 정수형이어야 함
 
 # 고정된 네거티브 프롬프트
 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"
 
-# 디바이스 설정
-device = "cuda" if torch.cuda.is_available() else "cpu"
+# 디바이스 설정 (CPU 사용)
+device = "cpu"
 
-# 이미지 생성 파이프라인 설정
-pipeline = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float32)
+# Hugging Face 인증
+hf_token = os.getenv('HF_TOKEN')
+
+# 이미지 생성 파이프라인 설정 (CPU 사용 및 Hugging Face 토큰 사용)
+pipeline = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-Final", torch_dtype=torch.float32, use_auth_token=hf_token)
 pipeline = pipeline.to(device)
 
 # 프롬프트 번역 함수
@@ -99,3 +101,4 @@ if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(discord_token)
 
+