Li commited on
Commit
7c708d1
1 Parent(s): c941456

add transformers

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. transformers +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  os.system("cd open_flamingo && pip install .")
 
3
  import numpy as np
4
  import torch
5
  from PIL import Image
@@ -99,7 +100,7 @@ def generate(
99
  height = image.height
100
  image = image.resize((224, 224))
101
  batch_images = image_processor(image).unsqueeze(0).unsqueeze(1).unsqueeze(0)
102
- if idx ==1:
103
  prompt = [f"<|#image#|>{tokenizer.pad_token*vis_embed_size}<|#endofimage#|><|#obj#|>{text.rstrip('.')}"]
104
  bad_words_ids = bad_words_ids
105
  else:
 
1
  import os
2
  os.system("cd open_flamingo && pip install .")
3
+ os.system("cd transformers && pip install .")
4
  import numpy as np
5
  import torch
6
  from PIL import Image
 
100
  height = image.height
101
  image = image.resize((224, 224))
102
  batch_images = image_processor(image).unsqueeze(0).unsqueeze(1).unsqueeze(0)
103
+ if idx == 1:
104
  prompt = [f"<|#image#|>{tokenizer.pad_token*vis_embed_size}<|#endofimage#|><|#obj#|>{text.rstrip('.')}"]
105
  bad_words_ids = bad_words_ids
106
  else:
transformers ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit 9417c924af539be5f941c8a709a96b60dfe29eb3