WrittenRec / app.py
JustKiddo's picture
Update app.py
0bb8b0b verified
raw
history blame
872 Bytes
import gradio as gr
import numpy as np
from huggingface_hub import InferenceClient
client = InferenceClient("models/microsoft/trocr-base-handwritten")
def sepia(input_img):
sepia_filter = np.array([
[0.393, 0.769, 0.189],
[0.349, 0.686, 0.168],
[0.272, 0.534, 0.131]
])
sepia_img = input_img.dot(sepia_filter.T)
sepia_img /= sepia_img.max()
return sepia_img
sepia_interface = gr.Interface(sepia, gr.Image(), "image")
with gr.Blocks() as generated_output:
gr.Markdown(
"""
# Hello World!
Start typing below to see the output.
""")
inp = gr.Textbox(placeholder="What is your name?")
out = gr.Textbox()
inp.change(welcome, inp, out)
demoApp = gr.TabbedInterface([sepia_interface, generated_output], ["RGB Sepia Filter", "Handwritten to Text"])
if __name__ == "__main__":
demoApp.launch()