Update tasks/text.py

tasks/text.py

@@ -21,18 +21,19 @@ ROUTE = "/text"
 @router.post(ROUTE, tags=["Text Task"], 
              description=DESCRIPTION)
 
-#modified: retrieve model
-model_repo = "jennasparks/electra_tf"
-config = ElectraConfig.from_pretrained(model_repo)
-model = TFElectraForSequenceClassification.from_pretrained(model_repo)
-tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
-#
-
 async def evaluate_text(request: TextEvaluationRequest):
     """
     Evaluate text classification for climate disinformation detection.
     Current Model: Finetuned ELECTRA
     """
+
+    #modified: retrieve model
+    model_repo = "jennasparks/electra_tf"
+    config = ElectraConfig.from_pretrained(model_repo)
+    model = TFElectraForSequenceClassification.from_pretrained(model_repo)
+    tokenizer = ElectraTokenizer.from_pretrained("google/electra-base-discriminator")
+    #
+    
     # Get space info
     username, space_url = get_space_info()