Hugging Face's logo

Spaces:
SharkSpace
/
shark_detection
Runtime error

App Files Community
piperod commited on
Commit
cc37e2b
·
1 Parent(s): 780307f

updating reference repos

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. inference.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import subprocess
5
  from huggingface_hub import snapshot_download
6
 
7
 
8
- REPO_ID='piperod91/videos_examples'
9
  snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),repo_type='dataset',local_dir='videos_example')
10
 
11
 
 
5
  from huggingface_hub import snapshot_download
6
 
7
 
8
+ REPO_ID='SharkSpace/videos_examples'
9
  snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),repo_type='dataset',local_dir='videos_example')
10
 
11
 
inference.py CHANGED
@@ -73,7 +73,7 @@ classes= ['Beach',
73
 
74
 
75
 
76
- REPO_ID = "piperod91/australiapatrol"
77
  FILENAME = "mask2former"
78
 
79
  snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')
 
73
 
74
 
75
 
76
+ REPO_ID = "SharkSpace/maskformer_model"
77
  FILENAME = "mask2former"
78
 
79
  snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')