onnew commited on
Commit
af086ac
·
verified ·
1 Parent(s): e73c41b

Create The generated image will be saved as

Browse files
The generated image will be saved as ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ### Explicação das Correções
2
+
3
+ 1. **Aspas nas Cores**: As cores `colorFrom` e `colorTo` devem estar entre aspas para garantir que sejam interpretadas corretamente como strings.
4
+ 2. **Formatação YAML**: Certifique-se de que não há espaços extras ou caracteres inválidos na seção YAML.
5
+
6
+ ### Arquivos do Projeto
7
+
8
+ Certifique-se de que os outros arquivos (`app.py`, `live_preview_helpers.py`, `requirements.txt`) estejam configurados conforme mencionado anteriormente.
9
+
10
+ ### `app.py`
11
+
12
+ ```python
13
+ import requests
14
+ import json
15
+ from live_preview_helpers import preview_image
16
+
17
+ # Defina a URL da API do Hugging Face
18
+ API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
19
+
20
+ # Defina o cabeçalho da requisição com o token de autenticação
21
+ headers = {
22
+ "Authorization": "Bearer YOUR_HUGGINGFACE_API_TOKEN"
23
+ }
24
+
25
+ def query(payload):
26
+ response = requests.post(API_URL, headers=headers, json=payload)
27
+ return response
28
+
29
+ def generate_image(text_prompt, output_file="generated_image.png"):
30
+ data = {
31
+ "inputs": text_prompt,
32
+ "options": {
33
+ "wait_for_model": True,
34
+ "use_cpu": True,
35
+ "cpu_basic": {
36
+ "vCPU": 2,
37
+ "RAM": 16
38
+ }
39
+ }
40
+ }
41
+
42
+ response = query(data)
43
+
44
+ if response.status_code == 200:
45
+ with open(output_file, "wb") as f:
46
+ f.write(response.content)
47
+ print(f"Imagem gerada com sucesso e salva como '{output_file}'")
48
+ preview_image(output_file) # Chama a função de visualização ao vivo
49
+ else:
50
+ print(f"Erro na requisição: {response.status_code}")
51
+ print(f"Mensagem de erro: {response.text}")
52
+
53
+ if __name__ == "__main__":
54
+ # Defina o texto que será usado para gerar a imagem
55
+ text_prompt = "A beautiful sunset over the mountains"
56
+
57
+ # Gere a imagem
58
+ generate_image(text_prompt)