| import requests | |
| # Substitua pelo seu URL do ngrok ou Space | |
| url = "eeda-34-124-254-92.ngrok-free.app/generate" # Certifique-se de usar o endpoint correto | |
| def infer_simple(prompt): | |
| # Envia uma requisição POST para o endpoint do servidor | |
| response = requests.post(url, json={"prompt": prompt}) | |
| # Verifica se a requisição foi bem-sucedida | |
| if response.status_code == 200: | |
| return response.json()["result"] | |
| else: | |
| print(f"Erro: {response.status_code} - {response.text}") | |
| return None | |
| # Testando a função | |
| if __name__ == "__main__": | |
| prompt = "A beautiful brunette woman wearing a blue leather pants" | |
| result = infer_simple(prompt) | |
| print(f"Resultado: {result}") | |