xichen98cn commited on
Commit
5e46b82
·
verified ·
1 Parent(s): 88c6e84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -125,7 +125,7 @@ def build_demo_classes_and_metadata(vocab, label_list):
125
  demo_classes = demo_thing_classes + demo_stuff_classes
126
  return demo_classes, demo_metadata
127
 
128
- @spaces.GPU
129
  def inference(image_path, vocab, label_list):
130
  logger.info("building class names")
131
  vocab = vocab.replace(", ", ",").replace("; ", ";")
 
125
  demo_classes = demo_thing_classes + demo_stuff_classes
126
  return demo_classes, demo_metadata
127
 
128
+ @spaces.GPU(duration=120)
129
  def inference(image_path, vocab, label_list):
130
  logger.info("building class names")
131
  vocab = vocab.replace(", ", ",").replace("; ", ";")