Spaces:
Paused
Paused
Update app.py
Browse files
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(
|