Update tasks/text.py

tasks/text.py

@@ -62,18 +62,13 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     from transformers import AutoModelForSequenceClassification, AutoTokenizer
     import torch
-    from huggingface_hub import hf_hub_download
-
+  
     # Load model and tokenizer from Hugging Face Hub
     MODEL_REPO = "ClimateDebunk/FineTunedDistilBert4SeqClass"
-    MODEL_FILENAME = "distilbert_trained.pth"
 
-    model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILENAME)
-    
     tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', do_lower_case=True)
     model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO)
 
-    model.load_state_dict(torch.load(model_path, map_location=torch.device("cpu")))
     model.eval()  # Set to evaluation mode
     
     def preprocess(texts):