Spaces:
No application file
No application file
File size: 920 Bytes
8023e00 6996c97 8023e00 6996c97 8023e00 6996c97 8023e00 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import os
from huggingface_hub import InferenceClient
import gradio as gr
from dotenv import load_dotenv
load_dotenv()
HF_TOKEN = os.getenv("HF_TOKEN")
if HF_TOKEN is None:
raise ValueError("No se encontró la variable HF_TOKEN en el .env")
client = InferenceClient(provider="hf-inference", api_key=HF_TOKEN)
def generar_imagen(prompt):
if not prompt.strip():
return None
imagen = client.text_to_image(prompt, model="black-forest-labs/FLUX.1-schnell")
return imagen
with gr.Blocks() as demo:
gr.Markdown("#Generador de Imágenes")
gr.Markdown("Escribe un prompt y el modelo generará la imagen usando Inference.")
prompt_input = gr.Textbox(label="Escribe tu prompt aquí", lines=3)
boton = gr.Button("Generar Imagen")
imagen_output = gr.Image(label="Imagen Generada")
boton.click(fn=generar_imagen, inputs=prompt_input, outputs=imagen_output)
demo.launch()
|