Upload app.py

app.py

@@ -1,20 +1,8 @@
 import gradio as gr
 from gradio_litmodel3d import LitModel3D
+import spaces
 
 import os
-if os.environ.get("SPACES_ZERO_GPU") is not None:
-    import spaces
-else:
-    class spaces:
-        @staticmethod
-        def GPU(func):
-            def wrapper(*args, **kwargs):
-                return func(*args, **kwargs)
-            return wrapper
-
-@spaces.GPU
-def fake_gpu():
-  pass
 
 import shutil
 os.environ['SPCONV_ALGO'] = 'native'
@@ -35,7 +23,8 @@ import cv2
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
-
+torch.set_grad_enabled(False)
+device = torch.device('cuda')
 
 def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
@@ -555,4 +544,5 @@ if __name__ == "__main__":
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))
     except:
         pass
-    demo.launch()
+    demo.queue(max_size=1)
+    demo.launch(share=False)