changed repo intake
Browse files- tasks/text.py +0 -5
tasks/text.py
CHANGED
|
@@ -22,11 +22,6 @@ ROUTE = "/text"
|
|
| 22 |
# model_weights_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="tf_model.h5")
|
| 23 |
# model_config_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="config.json")
|
| 24 |
|
| 25 |
-
# config = ElectraConfig.from_json_file(model_config_path)
|
| 26 |
-
# model = TFElectraForSequenceClassification(config)
|
| 27 |
-
# model.load_weights(model_weights_path)
|
| 28 |
-
# tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
|
| 29 |
-
|
| 30 |
model_repo = "jennasparks/electra_tf"
|
| 31 |
config = ElectraConfig.from_pretrained(model_repo)
|
| 32 |
model = TFElectraForSequenceClassification.from_pretrained(model_repo)
|
|
|
|
| 22 |
# model_weights_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="tf_model.h5")
|
| 23 |
# model_config_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="config.json")
|
| 24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
model_repo = "jennasparks/electra_tf"
|
| 26 |
config = ElectraConfig.from_pretrained(model_repo)
|
| 27 |
model = TFElectraForSequenceClassification.from_pretrained(model_repo)
|