Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
['
|
8 |
-
['
|
9 |
-
['
|
10 |
-
['
|
11 |
-
['
|
12 |
-
['
|
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')
|