model loading change

tasks/text.py

@@ -18,14 +18,19 @@ router = APIRouter()
 DESCRIPTION = "Electra_Base"
 ROUTE = "/text"
 
-# Load model and tokenizer
-model_weights_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="tf_model.h5")
-model_config_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="config.json")
-
-config = ElectraConfig.from_json_file(model_config_path)
-model = TFElectraForSequenceClassification(config)
-model.load_weights(model_weights_path)
-tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
+# # Load model and tokenizer
+# model_weights_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="tf_model.h5")
+# model_config_path = hf_hub_download(repo_id="jennasparks/electra-tf", filename="config.json")
+
+# config = ElectraConfig.from_json_file(model_config_path)
+# model = TFElectraForSequenceClassification(config)
+# model.load_weights(model_weights_path)
+# tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
+
+model_repo = "jennasparks/electra-tf"
+config = AutoConfig.from_pretrained(model_repo)
+model = TFAutoModelForSequenceClassification.from_pretrained(model_repo)
+tokenizer = AutoTokenizer.from_pretrained(model_repo)
 
 
 @router.post(ROUTE, tags=["Text Task"],