anisotropies commited on
Commit
b80f28f
·
verified ·
1 Parent(s): 070c1a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
 
 
3
  def yes_man(message, history):
4
  # if message.endswith("?"):
5
  # return "Yes"
@@ -7,7 +8,7 @@ def yes_man(message, history):
7
  # return "Ask me anything!"
8
 
9
  import requests
10
- headers = {"Authorization": f"Bearer {HF_TOKEN}"}
11
  API_URL = "https://api-inference.huggingface.co/models/gpt2"
12
  def query(payload):
13
  response = requests.post(API_URL, headers=headers, json=payload)
 
1
  import gradio as gr
2
 
3
+ import os
4
  def yes_man(message, history):
5
  # if message.endswith("?"):
6
  # return "Yes"
 
8
  # return "Ask me anything!"
9
 
10
  import requests
11
+ headers = {"Authorization": f"Bearer {os.getenv("HF_TOKEN")}"}
12
  API_URL = "https://api-inference.huggingface.co/models/gpt2"
13
  def query(payload):
14
  response = requests.post(API_URL, headers=headers, json=payload)