Update app.py

app.py

@@ -168,7 +168,8 @@ sam_checkpoint_url_dict = {
     'vit_l': "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth",
     'vit_b': "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth"
 }
-checkpoint_folder = os.path.join('/home/user/app/', 'pretrained_models')
+#checkpoint_folder = os.path.join('/home/user/app/', 'pretrained_models')
+checkpoint_folder = os.path.join('.', 'pretrained_models')
 
 sam_checkpoint = load_file_from_url(sam_checkpoint_url_dict[args.sam_model_type], checkpoint_folder)
 # initialize sams
@@ -239,8 +240,13 @@ html_img = f'''
 </div>
 '''
 
+#checkpoint_folder = os.path.join('/home/user/app/', 'pretrained_models')
+#checkpoint_folder = os.path.join('.', 'pretrained_models')
+
+
 tutorial_url = "https://github.com/zjx0101/ObjectClear/releases/download/media/tutorial.mp4"
-assets_path = os.path.join('/home/user/app/hugging_face/', "assets/")
+#assets_path = os.path.join('/home/user/app/hugging_face/', "assets/")
+assets_path = os.path.join('hugging_face/', "assets/")
 load_file_from_url(tutorial_url, assets_path)
 
 description = r"""
@@ -564,4 +570,4 @@ with gr.Blocks(css=custom_css) as demo:
     )
 
 
-demo.launch(debug=True, show_error=True)
+demo.launch(debug=True, show_error=True, share = True, server_name = "0.0.0.0")