WrittenRec / app.py
JustKiddo's picture
Update app.py
9d478e6 verified
raw
history blame
998 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")
## https://www.gradio.app/docs/gradio/blocks
## required positional arguments: 'inputs' and 'outputs'
generated_output = gr.Interface(None,
inputs = gr.Image("image"),
outputs = gr.Textbox())
#with gr.Blocks() as generated_output:
# inp = gr.Interface(sepia, gr.Image(), "image")
# out = gr.Textbox()
demoApp = gr.TabbedInterface([sepia_interface, generated_output], ["RGB Sepia Filter", "Handwritten to Text"])
if __name__ == "__main__":
demoApp.launch()