Remove unsqueeze for semantic tokens

app.py

@@ -92,7 +92,7 @@ class Predictor(object):
         concepts, scores = self.model.predict_concept(outputs["sem_embeds"][mask_index])
         concepts, scores = [x.reshape(batch_shape) for x in (concepts, scores)]
         # Generate captions.
-        sem_tokens = outputs["sem_tokens"][mask_index].unsqueeze_(1)
+        sem_tokens = outputs["sem_tokens"][mask_index]
         captions = self.model.generate_text(sem_tokens).reshape(batch_shape)
         # Postprocess results.
         results = []