Update app.py

app.py

@@ -451,10 +451,9 @@ with block:
                     outputs=[result, preview, progress_desc, progress_bar, start_btn, stop_btn])
     stop_btn.click(fn=end_process)
 
-# アプリ起動
-type(block.launch())
-', torch_dtype=transformer_dtype
-        ).to('cpu')
+    # アプリ起動
+    device = 'cuda' if GPU_AVAILABLE and not cpu_fallback_mode else 'cpu'
+    print(f"使用デバイス: {device} を推論に使います")
 
         # 評価モードに設定
         vae.eval(); text_encoder.eval(); text_encoder_2.eval(); image_encoder.eval(); transformer.eval()