Update inference.py

inference.py

@@ -1,6 +1,6 @@
 import torch
 import json
-import requests
+from huggingface_hub import hf_hub_download
 import re
 import emoji
 from transformers import BertForSequenceClassification, BertTokenizer
@@ -24,9 +24,10 @@ def load_model_and_resources():
     except Exception as e:
         raise RuntimeError(f"Error loading model/tokenizer: {str(e)}")
 
-    thresholds_url = f"https://huggingface.co/{repo_id}/raw/main/optimized_thresholds.json"
     try:
-        thresholds_data = json.loads(requests.get(thresholds_url).text)
+        thresholds_file = hf_hub_download(repo_id=repo_id, filename="optimized_thresholds.json")
+        with open(thresholds_file, "r") as f:
+            thresholds_data = json.load(f)
         if not (isinstance(thresholds_data, dict) and "emotion_labels" in thresholds_data and "thresholds" in thresholds_data):
             raise ValueError("Unexpected format in optimized_thresholds.json. Expected a dictionary with keys 'emotion_labels' and 'thresholds'.")
         emotion_labels = thresholds_data["emotion_labels"]