Update app.py

app.py

@@ -9,6 +9,9 @@ import subprocess
 import threading
 import torch
 from subprocess import getoutput
+import torch  
+torch.cuda.set_device(1)  # GPU 1번 사용
+torch.backends.cudnn.enabled = True
 
 # 환경 변수 대신 코드 내에서 직접 설정
 is_shared_ui = False  # 또는 True, 필요에 따라 설정