aiqtech commited on
Commit
40669db
·
verified ·
1 Parent(s): 3df02e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -440,8 +440,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
440
 
441
  if __name__ == "__main__":
442
  try:
443
- # CPU 모드로 초기화
444
- device = "cpu"
445
  print(f"Using device: {device}")
446
 
447
  # 모델 초기화
@@ -461,7 +461,7 @@ if __name__ == "__main__":
461
  server_name="0.0.0.0",
462
  server_port=7860,
463
  show_error=True,
464
- share=False
465
  )
466
 
467
  except Exception as e:
 
440
 
441
  if __name__ == "__main__":
442
  try:
443
+ # CUDA 사용 가능 여부 확인
444
+ device = "cuda" if torch.cuda.is_available() else "cpu"
445
  print(f"Using device: {device}")
446
 
447
  # 모델 초기화
 
461
  server_name="0.0.0.0",
462
  server_port=7860,
463
  show_error=True,
464
+ share=True # share를 True로 설정
465
  )
466
 
467
  except Exception as e: