amosyou commited on
Commit
7a6408f
1 Parent(s): 53b5c44

feat: include examples

Browse files
app.py CHANGED
@@ -44,10 +44,15 @@ with gr.Blocks() as demo:
44
  **Model**: [ditto](https://huggingface.co/Launchpad/ditto)
45
  <br/>
46
  **Developed by**: Kiran Suresh, Annie Lee, Chloe Wong, Tony Xin, Sebastian Zhao
 
 
47
  """
48
  )
49
- with gr.Row():
50
- gr.Interface(generate_pokemon, gr.Image(), "image")
 
 
 
51
 
52
  if __name__ == '__main__':
53
  demo.launch()
 
44
  **Model**: [ditto](https://huggingface.co/Launchpad/ditto)
45
  <br/>
46
  **Developed by**: Kiran Suresh, Annie Lee, Chloe Wong, Tony Xin, Sebastian Zhao
47
+ <br/>
48
+ **Examples**: [Oxford-IIIT Pet Dataset](https://www.robots.ox.ac.uk/~vgg/data/pets/)
49
  """
50
  )
51
+ gr.Interface(fn=generate_pokemon,
52
+ inputs=gr.Image(),
53
+ outputs="image",
54
+ examples=["data/german_shorthaired_164.jpg", "data/samoyed_189.jpg", "data/shiba_inu_139.jpg"]
55
+ )
56
 
57
  if __name__ == '__main__':
58
  demo.launch()
data/german_shorthaired_164.jpg ADDED
data/samoyed_189.jpg ADDED
data/shiba_inu_139.jpg ADDED