fantaxy commited on
Commit
b452fe3
1 Parent(s): f218991

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -58,8 +58,7 @@ model.eval()
58
  model = model.cuda()
59
 
60
  # 번역 모델 로드
61
- translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=0 if torch.cuda.is_available() else -1)
62
-
63
  # 이미지 생성 모델 로드
64
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
65
  pipe = StableDiffusionXLPipeline.from_pretrained(
 
58
  model = model.cuda()
59
 
60
  # 번역 모델 로드
61
+ translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=0 if torch.cuda.is_available() else -1, framework="pt")
 
62
  # 이미지 생성 모델 로드
63
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
64
  pipe = StableDiffusionXLPipeline.from_pretrained(