Views
No views yet
1import requests
2
3API_URL = "https://YOUR-ENDPOINT-URL"
4headers = {"Authorization": "Bearer YOUR_HF_TOKEN"}
5
6def query(payload):
7 response = requests.post(API_URL, headers=headers, json=payload)
8 return response.json()
9
10output = query({
11 "inputs": "Write a short story about a robot learning to paint:",
12 "parameters": {
13 "max_new_tokens": 250,
14 "temperature": 0.7,
15 "top_p": 0.95,
16 "do_sample": true
17 }
18})1api_inference:
2 handler_class: handler.EndpointHandler