ExhibitAI / app.py
dalybuilds's picture
Create app.py
c0c6362 verified
raw
history blame
659 Bytes
import gradio as gr
from datasets import load_dataset
# Load the WikiArt dataset
dataset = load_dataset("huggan/wikiart")
# Function to display artwork details
def display_artwork(index):
record = dataset["train"][index]
return record["image"], f"Welcome to the Gallery\n\nTitle: {record['title']}\nArtist: {record['artist']}\nStyle: {record['style']}\nGenre: {record['genre']}\n\nStep into the world of art and explore its details."
# Gradio Interface
gr.Interface(
fn=display_artwork,
inputs=gr.Number(label="Artwork Index"),
outputs=[gr.Image(label="Artwork"), gr.Text(label="Details")],
title="Virtual AI Art Gallery"
).launch()