import gradio as gr
from huggingface_hub import InferenceClient

client = InferenceClient("BillyAggarwal/NeuralStyleTrasnferModel.keras")

def style_transfer(content_image, style_image):
    # send images to Hugging Face model
    result = client.post(
        json={"inputs": {"content": content_image, "style": style_image}}
    )
    return result

demo = gr.Interface(
    fn=style_transfer,
    inputs=[gr.Image(), gr.Image()],
    outputs=gr.Image(),
    title="Neural Style Transfer"
)

demo.launch()