Chris4K commited on
Commit
d4221fc
1 Parent(s): 1fad4a0

Update text_generator.py

Browse files
Files changed (1) hide show
  1. text_generator.py +1 -1
text_generator.py CHANGED
@@ -17,7 +17,7 @@ def __call__(self, prompt: str):
17
 
18
 
19
  API_URL = "https://api-inference.huggingface.co/models/lukasdrg/clinical_longformer_same_tokens_220k"
20
- headers = {"Authorization": "Bearer "+os.environ['HF']+"}
21
 
22
  #def query(payload):
23
  generated_text = requests.post(API_URL, headers=headers, json=payload)
 
17
 
18
 
19
  API_URL = "https://api-inference.huggingface.co/models/lukasdrg/clinical_longformer_same_tokens_220k"
20
+ headers = {"Authorization": "Bearer "+os.environ['HF']}
21
 
22
  #def query(payload):
23
  generated_text = requests.post(API_URL, headers=headers, json=payload)