re-add gdown
Browse files- tasks/text.py +5 -5
tasks/text.py
CHANGED
|
@@ -20,12 +20,12 @@ ROUTE = "/text"
|
|
| 20 |
- Makes random predictions from the label space (0-7)
|
| 21 |
- Used as a baseline for comparison
|
| 22 |
"""
|
| 23 |
-
#
|
| 24 |
-
|
| 25 |
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
|
| 30 |
model = tf.keras.models.load_model('checkpoint_epoch_5.weights.h5')
|
| 31 |
|
|
|
|
| 20 |
- Makes random predictions from the label space (0-7)
|
| 21 |
- Used as a baseline for comparison
|
| 22 |
"""
|
| 23 |
+
# Download from Google Drive
|
| 24 |
+
import gdown
|
| 25 |
|
| 26 |
+
url = 'https://drive.google.com/uc?id=1-HWE2G6ANbd7mqILdB9DPrvF3DrKI1e4'
|
| 27 |
+
output = 'checkpoint_epoch_5.weights.h5'
|
| 28 |
+
gdown.download(url, output, quiet=False)
|
| 29 |
|
| 30 |
model = tf.keras.models.load_model('checkpoint_epoch_5.weights.h5')
|
| 31 |
|