NeuralStyleTransfer / gradioapp
BillyAggarwal's picture
Update gradioapp
8c7bfb0 verified
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()