File size: 2,554 Bytes
8ea9424 989fb86 c8f46e3 989fb86 e4d5724 ff50a6e 6086f9a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | import os
os.system('pip install -r requirements-colab.txt')
os.system('pip install gradio tensorflow numpy')
from deoldify import device
from deoldify.device_id import DeviceId
device.set(device=DeviceId.GPU0)
import torch
if not torch.cuda.is_available():
print('GPU not available.')
import fastai
from deoldify.visualize import *
import warnings
warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
colorizer = get_image_colorizer(artistic=True)
# source_url = 'https://i.pinimg.com/originals/09/c2/22/09c222cc275efa908bdc140a750ca4e4.jpg'
# image_path_1 = colorizer.plot_transformed_image_from_url(url=source_url, render_factor=render_factor, compare=True, watermarked=watermarked)
# image_path_2 = colorizer.plot_transformed_image(path=source_path, render_factor=render_factor, compare=True)
# print("image_path",image_path_1)
render_factor=10
import gradio as gr
import tensorflow as tf
import numpy as np
import os
def Colorize(address,watermark,URL,render_factor):
text_box_output = "Successful"
watermarked = True
output_address = np.array([
[0,0],
[0,0],
], dtype=np.uint8)
if watermark == "Watermark On":
watermarked = True
render_factor = int(render_factor)
elif watermark == "Watermark Off":
watermarked = False
render_factor = int(render_factor)
else:
text_box_output = "Please set the watermark setting."
if URL != '':
output_address = colorizer.plot_transformed_image_from_url(url=URL, render_factor=render_factor, compare=True, watermarked=watermarked)
else:
if address == None:
text_box_output = "Please provide the image"
else:
output_address = colorizer.plot_transformed_image(path=address, render_factor=render_factor, compare=True, watermarked=watermarked)
return text_box_output,output_address
with gr.Blocks() as demo:
image_box=gr.Image(type="filepath")
image_output_box = gr.Image()
# text_box_1 = gr.Textbox(value="10",label="The render factor")
text_box_2 = gr.Textbox(label='URL')
drop_down_1 = gr.Dropdown(['Watermark On','Watermark Off'],label='WaterMark')
slider = gr.Slider(7,40,step=1)
text_box_output = gr.Textbox(label="Status")
slider.release(Colorize, inputs=[image_box,drop_down_1,text_box_2,slider], outputs=[text_box_output,image_output_box])
#b1=gr.Button("Colorized!")
#b1.click(Colorize, inputs=[image_box,drop_down_1,text_box_2,text_box_1], outputs= image_output_box)
demo.queue().launch(share=True,debug=True) |