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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import requests
2
  import json
 
3
 
4
  # Defina a URL da API do Hugging Face
5
  API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
@@ -32,6 +33,7 @@ def generate_image(text_prompt, output_file="generated_image.png"):
32
  with open(output_file, "wb") as f:
33
  f.write(response.content)
34
  print(f"Imagem gerada com sucesso e salva como '{output_file}'")
 
35
  else:
36
  print(f"Erro na requisição: {response.status_code}")
37
  print(f"Mensagem de erro: {response.text}")
 
1
  import requests
2
  import json
3
+ from live_preview_helpers import preview_image
4
 
5
  # Defina a URL da API do Hugging Face
6
  API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
 
33
  with open(output_file, "wb") as f:
34
  f.write(response.content)
35
  print(f"Imagem gerada com sucesso e salva como '{output_file}'")
36
+ preview_image(output_file) # Chama a função de visualização ao vivo
37
  else:
38
  print(f"Erro na requisição: {response.status_code}")
39
  print(f"Mensagem de erro: {response.text}")