Update demo.py

demo.py

@@ -27,8 +27,8 @@ import requests
 from datetime import datetime
 import random
 
-# 번역 파이프라인 생성
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+# 파이프라인에 device 인자 추가하여 GPU 설정
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device="cuda")
 
 # 번역 함수
 def translate_prompt(korean_prompt):