Update app.py

app.py

@@ -1,18 +1,21 @@
 import gradio as gr
 import requests
+from PIL import Image
+from io import BytesIO
 
 # Função para gerar imagem usando a API do Hugging Face
 def generate_image(prompt: str):
     API_URL = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
-    headers = {"Authorization": "Bearer YOUR_HUGGINGFACE_API_KEY"}
+    headers = {"Authorization": f"Bearer YOUR_HUGGINGFACE_API_KEY"}
 
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
 
     if response.status_code == 200:
-        image_url = response.json()
-        return image_url
+        image_data = response.content
+        image = Image.open(BytesIO(image_data))
+        return image
     else:
-        return None
+        raise gr.Error(f"Erro ao gerar imagem: {response.status_code} - {response.text}")
 
 # Função para criar a interface do Gradio
 def gradio_interface():