BardsTale / app.py
NeuralJunkie's picture
Update app.py
797352a
raw
history blame contribute delete
484 Bytes
import requests
def get_user_input(url):
requests.packages.urllib3.disable_warnings()
response = requests.get(url)
return response.text
def main():
url = "https://api.huggingface.co/models/bard/generate"
headers = {
"Authorization": "Bearer YOUR_TOKEN"
}
data = {
"text": get_user_input(url)
}
response = requests.post(url, headers=headers, data=data)
print(response["generated_text"])
if __name__ == "__main__":
main()