Update app.py

app.py

@@ -706,13 +706,16 @@ def change_radio_display(task_type, mask_source_radio):
     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
+    try:
+        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
+    except Exception as e:
+        return 'ohoh'
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser("Grounded SAM demo", add_help=True)