Update gradio_app.py

gradio_app.py

@@ -89,6 +89,7 @@ def chmod(path, mode):
 
 chmod(f"{parent_dir}/apps/third_party/InstantMeshes", "777")
 
+device = None
 model = None
 cached_dir = None
 generator = None
@@ -108,7 +109,8 @@ def gen_mvimg(
 ):
     if seed == 0:
         seed = np.random.randint(1, 65535)
-        global generator
+        global generatorm device
+        generator = torch.Generator(device)
         generator.manual_seed(seed)
 
     if mvimg_model == "CRM":
@@ -277,7 +279,6 @@ if __name__=="__main__":
             trust_remote_code=True,
         )
 
-    generator = torch.Generator(device)
 
     # for 3D latent set diffusion
     ckpt_path = hf_hub_download(repo_id="wyysf/CraftsMan", filename="image-to-shape-diffusion/clip-mvrgb-modln-l256-e64-ne8-nd16-nl6-aligned-vae/model.ckpt", repo_type="model")