| import numpy as np | |
| import torch | |
| import torch.nn as nn | |
| import gradio as gr | |
| 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 | |
| iface = gr.Interface(sepia, gr.inputs.Image(shape=(256, 256)), "image") | |
| iface.launch() |