Caroline Mai Chan commited on
Commit
764782e
1 Parent(s): 4f625a2

add title, example

Browse files
Files changed (2) hide show
  1. app.py +13 -1
  2. cat.png +0 -0
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