autonomous019 commited on
Commit
fd787f5
1 Parent(s): fe25003

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, 32, "Sample", 42)
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
- #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>"
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>"