Update app.py

app.py

@@ -1,4 +1,11 @@
 import os
+import spaces
+import gradio as gr
+import random
+import numpy as np
+import torch
+from torchvision.transforms.functional import center_crop
+
 try:
     # Try to install detectron2 from source. Needed for semseg plotting functionality.
     os.system("python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'")
@@ -6,7 +13,7 @@ except Exception as e:
     print('detectron2 cannot be installed. Falling back to simple semseg visualization.')
     print(e)
 
-import torch
+
 # We recommend running this demo on an A100 GPU
 if torch.cuda.is_available():
     device = "cuda"
@@ -18,11 +25,6 @@ else:
     power_device = "CPU"
     os.system("pip uninstall -y xformers") # Only use xformers on GPU
 
-import spaces
-import gradio as gr
-import random
-import numpy as np
-from torchvision.transforms.functional import center_crop
 from fourm.demo_4M_sampler import Demo4MSampler
 from fourm.data.modality_transforms import RGBTransform