Spaces:
Runtime error
Runtime error
Li
commited on
Commit
·
7c708d1
1
Parent(s):
c941456
add transformers
Browse files- app.py +2 -1
- 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
|