add transformers

app.py

@@ -101,8 +101,8 @@ def generate(
     image = image.resize((224, 224))
     batch_images = image_processor(image).unsqueeze(0).unsqueeze(1).unsqueeze(0)
     if idx == 1:
-        prompt = [f"<|#image#|>{tokenizer.pad_token*vis_embed_size}<|#endofimage#|><|#obj#|>{text.rstrip('.')}"]
-        bad_words_ids = bad_words_ids
+        prompt = [f"<|#image#|>{tokenizer.pad_token*vis_embed_size}<|#endofimage#|><|#obj#|>{text.rstrip('.')}<|#loc#|>"]
+        bad_words_ids = None
     else:
         prompt = [f"<|#image#|>{tokenizer.pad_token * vis_embed_size}<|#endofimage#|>{text.rstrip('.')}"]
         bad_words_ids = None