acecalisto3 commited on
Commit
7d00a73
1 Parent(s): 85dfba5

Update tokenizer.py

Browse files
Files changed (1) hide show
  1. tokenizer.py +3 -0
tokenizer.py CHANGED
@@ -1,4 +1,7 @@
1
  from transformers import pipeline
 
 
 
2
 
3
  def generate_code_from_model(input_text):
4
  """Generates code using the bigscience/T0_3B model."""
 
1
  from transformers import pipeline
2
+ from huggingface_hub import hf_hub_url, cached_download, HfApi
3
+
4
+ HF_TOKEN = os.environ.get("HF_TOKEN", None)
5
 
6
  def generate_code_from_model(input_text):
7
  """Generates code using the bigscience/T0_3B model."""