imageNew / app.py
tejani's picture
Update app.py
db66451 verified
raw
history blame contribute delete
368 Bytes
from transformers import DiffusionPipeline # Example for diffusion models
import gradio as gr
model = DiffusionPipeline.from_pretrained("tejani/TileTexture")
def generate_image(text):
image = model(text).images[0] # Adjust based on output structure
return image
interface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
interface.launch()