Caroline Mai Chan
commited on
Commit
•
764782e
1
Parent(s):
4f625a2
add title, example
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ import torch
|
|
3 |
import torch.nn as nn
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
|
|
6 |
|
7 |
norm_layer = nn.InstanceNorm2d
|
8 |
|
@@ -86,13 +87,24 @@ model = Generator(3, 1, 3)
|
|
86 |
model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
|
87 |
model.eval()
|
88 |
|
|
|
|
|
|
|
|
|
89 |
def predict(input_img):
|
90 |
input_img = Image.open(input_img)
|
91 |
ratio = 256.0 / input_img.size[1]
|
92 |
input_img = input_img.resize((int(ratio*input_img.size[0]),256), Image.ANTIALIAS)
|
|
|
|
|
93 |
return input_img
|
94 |
|
|
|
|
|
|
|
|
|
|
|
95 |
|
96 |
-
iface = gr.Interface(predict, gr.inputs.Image(type='filepath'), "image")
|
97 |
|
98 |
iface.launch()
|
|
|
3 |
import torch.nn as nn
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
6 |
+
import torchvision.transforms as transforms
|
7 |
|
8 |
norm_layer = nn.InstanceNorm2d
|
9 |
|
|
|
87 |
model.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
|
88 |
model.eval()
|
89 |
|
90 |
+
transforms_r = [transforms.Resize(int(opt.size), Image.BICUBIC),
|
91 |
+
transforms.ToTensor()]
|
92 |
+
transforms.Compose([transforms.Resize(int(opt.size), Image.BICUBIC), transforms.ToTensor()])
|
93 |
+
|
94 |
def predict(input_img):
|
95 |
input_img = Image.open(input_img)
|
96 |
ratio = 256.0 / input_img.size[1]
|
97 |
input_img = input_img.resize((int(ratio*input_img.size[0]),256), Image.ANTIALIAS)
|
98 |
+
|
99 |
+
|
100 |
return input_img
|
101 |
|
102 |
+
title="informative-drawings"
|
103 |
+
description="Gradio Demo for line drawing generation. "
|
104 |
+
article = "<p style='text-align: center'><a href='https://film-net.github.io/' target='_blank'>FILM: Frame Interpolation for Large Motion</a> | <a href='https://github.com/google-research/frame-interpolation' target='_blank'>Github Repo</a></p>"
|
105 |
+
examples=[['cat.png']]
|
106 |
+
|
107 |
|
108 |
+
iface = gr.Interface(predict, gr.inputs.Image(type='filepath'), "image", title=title,description=description,article=article,examples=examples)
|
109 |
|
110 |
iface.launch()
|
cat.png
ADDED