download our model
Browse files- tasks/text.py +5 -0
tasks/text.py
CHANGED
|
@@ -42,6 +42,11 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 42 |
"7_fossil_fuels_needed": 7
|
| 43 |
}
|
| 44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
# Load and prepare the dataset
|
| 46 |
dataset = load_dataset(request.dataset_name)
|
| 47 |
|
|
|
|
| 42 |
"7_fossil_fuels_needed": 7
|
| 43 |
}
|
| 44 |
|
| 45 |
+
# Download our pre-trained model
|
| 46 |
+
url = 'https://drive.google.com/uc?id=1-HWE2G6ANbd7mqILdB9DPrvF3DrKI1e4'
|
| 47 |
+
output = 'checkpoint_epoch_5.weights.h5'
|
| 48 |
+
gdown.download(url, output, quiet=False)
|
| 49 |
+
|
| 50 |
# Load and prepare the dataset
|
| 51 |
dataset = load_dataset(request.dataset_name)
|
| 52 |
|