app.py
CHANGED
|
@@ -3,7 +3,9 @@ import requests
|
|
| 3 |
|
| 4 |
# GPT-J-6B API
|
| 5 |
API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
|
| 6 |
-
|
|
|
|
|
|
|
| 7 |
prompt = """
|
| 8 |
word: risk
|
| 9 |
poem using word: And then the day came,
|
|
|
|
| 3 |
|
| 4 |
# GPT-J-6B API
|
| 5 |
API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
|
| 6 |
+
HF_TOKEN = os.environ["HF_TOKEN"]
|
| 7 |
+
headers = {"Authorization": f"Bearer {HF_TOKEN}"}
|
| 8 |
+
#headers = {"Authorization": "Bearer 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"}
|
| 9 |
prompt = """
|
| 10 |
word: risk
|
| 11 |
poem using word: And then the day came,
|