in llm_swarm/__init__.py [0:0]
def test_generation(endpoint):
headers = {
"Content-Type": "application/json",
}
data = {
"inputs": "What is Deep Learning?",
"parameters": {
"max_new_tokens": 200,
},
}
requests.post(endpoint, headers=headers, json=data)
print("✅ test generation")