Update generators/model.py

generators/model.py

@@ -18,9 +18,10 @@ def model(type, template, seq_len=250):
         train = prompt.TOPIC_TO_ABSTRACST
 
     HF_URL = "https://api-inference.huggingface.co/models/bigscience/bloom"
-    HF_KEY = os.environ["HF_KEY"]
+    #HF_KEY = os.environ["HF_KEY"]
+    HF_TOKEN = os.environ["HF_TOKEN"]
 
-    headers = {"Authorization": f"Bearer {HF_KEY}"}
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
     print(f"Inside model {seq_len}")