Update app.py

app.py

@@ -3,6 +3,8 @@ import os
 import subprocess
 import shlex
 import spaces
+import torch
+import numpy as np
 subprocess.run(
     shlex.split(
         "pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt210/download.html"
@@ -48,8 +50,7 @@ def check_gpu():
     subprocess.run(['nvidia-smi'])  # 测试 CUDA 是否可用
     print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
 check_gpu()
-import torch
-import numpy as np
+
 from PIL import Image
 from einops import rearrange
 from diffusers import FluxPipeline