savtadepth / app /app_savta.py
Abid
requrement corrected
5dc6bd9
raw
history blame
1.37 kB
import numpy as np
import gradio as gr
title = "SavtaDepth WebApp"
description = "Savta Depth is a collaborative Open Source Data Science project for monocular depth estimation - Turn 2d photos into 3d photos. To test the model and code please check out the link bellow."
article = "<p style='text-align: center'><a href='https://dagshub.com/OperationSavta/SavtaDepth' target='_blank'>SavtaDepth Project from OperationSavta</a></p><p style='text-align: center'><a href='https://colab.research.google.com/drive/1XU4DgQ217_hUMU1dllppeQNw3pTRlHy1?usp=sharing' target='_blank'>Google Colab Demo</a></p></center></p>"
examples = [
["examples/00008.jpg"],
["examples/00045.jpg"],
]
favicon = "examples/favicon.ico"
thumbnail = "examples/SavtaDepth.png"
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
def main():
iface = gr.Interface(sepia, gr.inputs.Image(shape=(200, 200)), "image", title = title, description = description, article = article, examples = examples,theme ="peach",thumbnail=thumbnail)
iface.launch(favicon_path=favicon,auth=("admin", "dagshubsota123"),server_name="0.0.0.0")
# enable_queue=True,auth=("admin", "pass1234")
if __name__ == '__main__':
main()