Update app.py

app.py

@@ -705,6 +705,15 @@ def change_radio_display(task_type, mask_source_radio):
         num_relation_visible = True
     return  gr.Textbox.update(visible=text_prompt_visible), gr.Textbox.update(visible=inpaint_prompt_visible), gr.Radio.update(visible=mask_source_radio_visible), gr.Slider.update(visible=num_relation_visible) 
 
+def get_model_device(module):
+    if isinstance(module, torch.nn.DataParallel):
+        module = module.module
+    for submodule in module.children():
+        if hasattr(submodule, "_parameters"):
+            parameters = submodule._parameters
+            if "weight" in parameters:
+                return parameters["weight"].device
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser("Grounded SAM demo", add_help=True)
     parser.add_argument("--debug", action="store_true", help="using debug mode")
@@ -721,7 +730,12 @@ if __name__ == "__main__":
     load_ram_model()
 
     os.system("pip list")
-
+    print(f'groundingdino_model__{get_model_device(groundingdino_model)}')
+    print(f'sam_model__{get_model_device(sam_model)}')
+    print(f'sd_model__{get_model_device(sd_pipe)}')
+    print(f'lama_cleaner_model__{get_model_device(lama_cleaner_model)}')
+    print(f'ram_model__{get_model_device(ram_model)}')
+    
     block = gr.Blocks().queue()
     with block:
         with gr.Row():