File size: 368 Bytes
db66451
 
bb8e8f9
db66451
 
 
 
bb8e8f9
db66451
bb8e8f9
1
2
3
4
5
6
7
8
9
10
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()