Jiayi-Pan commited on
Commit
7e670b1
·
1 Parent(s): 19982c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -4,12 +4,12 @@ import gradio as gr
4
  title = "OctoBERT"
5
  description = """Interactive Demo for OctoBERT. This base model is trained only on Flickr-30k."""
6
  examples =[
7
- ['https://i.imgur.com/ASXtqVc.jpg','The woman stands outdoors, next to a child in a <mask>.'],
8
- ['https://i.imgur.com/UspUc7t.jpg', 'A woman in blue shorts and white shirt holds a tennis racket on a blue <mask> court.'],
9
- ['https://i.imgur.com/IRUoMUg.jpg', 'The smiling <mask> is celebrating her <mask> party with friends, surrounded by balloons and a <mask> with candles.'],
10
- ['https://i.imgur.com/23BZbaH.jpg', 'A person in a rainbow colored snowsuit is snowboarding down a <mask> slope.'],
11
- ['https://i.imgur.com/hoQCg1h.jpg', 'A man with <mask> plays with a little girl while walking down the street, while an Asian woman walks ahead of them.'],
12
- ['https://i.imgur.com/dFeN6bs.jpg', 'A black dog stands on a <mask>, green fields behind him.'],
13
  ]
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  model, img_transform, tokenizer, post_processor, plot_results = torch.hub.load('Jiayi-Pan/RefCloze_Pub', 'flickr_base_model')
 
4
  title = "OctoBERT"
5
  description = """Interactive Demo for OctoBERT. This base model is trained only on Flickr-30k."""
6
  examples =[
7
+ ['swing.jpg','The woman stands outdoors, next to a child in a <mask>.'],
8
+ ['tennis.jpg', 'A woman in blue shorts and white shirt holds a tennis racket on a blue <mask> court.'],
9
+ ['birthday.jpg', 'The smiling <mask> is celebrating her <mask> party with friends, surrounded by balloons and a <mask> with candles.'],
10
+ ['skate.jpg', 'A person in a rainbow colored snowsuit is snowboarding down a <mask> slope.'],
11
+ ['street.jpg', 'A man with <mask> plays with a little girl while walking down the street, while an Asian woman walks ahead of them.'],
12
+ ['dog.jpg', 'A black dog stands on a <mask>, green fields behind him.'],
13
  ]
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  model, img_transform, tokenizer, post_processor, plot_results = torch.hub.load('Jiayi-Pan/RefCloze_Pub', 'flickr_base_model')