liyuesen commited on
Commit
ee27ffc
1 Parent(s): 317fa29

Update drug_generator.py

Browse files
Files changed (1) hide show
  1. drug_generator.py +3 -5
drug_generator.py CHANGED
@@ -14,8 +14,9 @@ import torch
14
  from transformers import AutoTokenizer, GPT2LMHeadModel
15
 
16
  warnings.filterwarnings('ignore')
17
- os.environ["http_proxy"] = "http://127.0.0.1:7890"
18
- os.environ["https_proxy"] = "http://127.0.0.1:7890"
 
19
 
20
 
21
  # Set up command line argument parsing
@@ -138,6 +139,3 @@ for i in range(100):
138
 
139
 
140
 
141
-
142
-
143
-
 
14
  from transformers import AutoTokenizer, GPT2LMHeadModel
15
 
16
  warnings.filterwarnings('ignore')
17
+ #Sometimes, using Hugging Face may require a proxy.
18
+ #os.environ["http_proxy"] = "http://127.0.0.1:7890"
19
+ #os.environ["https_proxy"] = "http://127.0.0.1:7890"
20
 
21
 
22
  # Set up command line argument parsing
 
139
 
140
 
141