Update src/text_embedding.py

src/text_embedding.py

@@ -11,12 +11,12 @@ class TextExtractor:
         - model_name (str): The name of the pre-trained model to load from HuggingFace Hub.
         - proxy (str, optional): The proxy address to use for HTTP and HTTPS requests.
         """
-        if proxy is None:
-            proxy = 'http://localhost:8234'
+        # if proxy is None:
+        #     proxy = 'http://localhost:8234'
 
-        if proxy:
-            os.environ['HTTP_PROXY'] = proxy
-            os.environ['HTTPS_PROXY'] = proxy
+        # if proxy:
+        #     os.environ['HTTP_PROXY'] = proxy
+        #     os.environ['HTTPS_PROXY'] = proxy
         try:
             self.tokenizer = AutoTokenizer.from_pretrained(model_name)
             self.model = AutoModel.from_pretrained(model_name)