update

app.py

@@ -1,17 +1,14 @@
+import gradio as gr
+import spaces 
 import torch
 import torchvision
 from torchvision import transforms
 import numpy as np
 
 import os
-from omegaconf import OmegaConf
 from PIL import Image 
-
-import gradio as gr
-
 import rembg
 
-import spaces 
 
 from huggingface_hub import hf_hub_download
 
@@ -454,4 +451,14 @@ if __name__ == "__main__":
 
     args.rendering_kwargs = rendering_options_defaults(args)
 
-    main(args)
+    # main(args)
+
+    zero = torch.Tensor([0]).cuda()
+    print(zero.device) # <-- 'cpu' 🤔
+
+    @spaces.GPU
+    def greet(n):
+        print(zero.device) # <-- 'cuda:0' 🤗
+        return f"Hello {zero + n} Tensor"
+
+    demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())