Ved Gupta commited on
Commit
3c3c0de
·
1 Parent(s): 5ece346

GDRIVE download link added

Browse files
Files changed (2) hide show
  1. app/utils/utils.py +10 -1
  2. requirements.txt +2 -1
app/utils/utils.py CHANGED
@@ -1,8 +1,8 @@
1
- import json
2
  import subprocess
3
  import uuid
4
  import logging
5
  import wave
 
6
 
7
  from .constant import model_names
8
 
@@ -100,3 +100,12 @@ def get_model_name(model: str = None):
100
  return model_names[model]
101
 
102
  return model_names["tiny.en.q5"]
 
 
 
 
 
 
 
 
 
 
 
1
  import subprocess
2
  import uuid
3
  import logging
4
  import wave
5
+ import gdown
6
 
7
  from .constant import model_names
8
 
 
100
  return model_names[model]
101
 
102
  return model_names["tiny.en.q5"]
103
+
104
+
105
+ def download_from_drive(url , output):
106
+ try:
107
+ gdown.download(url, output, quiet=False)
108
+ return True
109
+ except:
110
+ print("Error Occured in Downloading model from Gdrive")
111
+ return False
requirements.txt CHANGED
@@ -10,4 +10,5 @@ requests-mock
10
  passlib
11
  httpx
12
  PyJWT
13
- python-multipart
 
 
10
  passlib
11
  httpx
12
  PyJWT
13
+ python-multipart
14
+ gdown