Spaces:
Build error
Build error
Commit
·
fd787f5
1
Parent(s):
fe25003
Update app.py
Browse files
app.py
CHANGED
|
@@ -117,7 +117,7 @@ def self_caption(image):
|
|
| 117 |
|
| 118 |
preds = ' '.join(preds)
|
| 119 |
#inference(input_sentence, max_length, sample_or_greedy, seed=42)
|
| 120 |
-
story = inference(preds,
|
| 121 |
|
| 122 |
return story
|
| 123 |
|
|
@@ -156,8 +156,8 @@ def classify_image(image):
|
|
| 156 |
|
| 157 |
image = gr.inputs.Image(type="pil")
|
| 158 |
label = gr.outputs.Label(num_top_classes=5)
|
| 159 |
-
|
| 160 |
-
examples = [ ["batter.jpg"] ]
|
| 161 |
title = "Generate a Story from an Image using BLOOM"
|
| 162 |
description = "Demo for classifying images with Perceiver IO. To use it, simply upload an image and click 'submit', a story is autogenerated as well, story generated using Bigscience/BLOOM"
|
| 163 |
article = "<p style='text-align: center'></p>"
|
|
|
|
| 117 |
|
| 118 |
preds = ' '.join(preds)
|
| 119 |
#inference(input_sentence, max_length, sample_or_greedy, seed=42)
|
| 120 |
+
story = inference(preds, 64, "Sample", 42)
|
| 121 |
|
| 122 |
return story
|
| 123 |
|
|
|
|
| 156 |
|
| 157 |
image = gr.inputs.Image(type="pil")
|
| 158 |
label = gr.outputs.Label(num_top_classes=5)
|
| 159 |
+
examples = [ ["cats.jpg"], ["batter.jpg"],["drinkers.jpg"] ]
|
| 160 |
+
#examples = [ ["batter.jpg"] ]
|
| 161 |
title = "Generate a Story from an Image using BLOOM"
|
| 162 |
description = "Demo for classifying images with Perceiver IO. To use it, simply upload an image and click 'submit', a story is autogenerated as well, story generated using Bigscience/BLOOM"
|
| 163 |
article = "<p style='text-align: center'></p>"
|