Spaces:
Build error
Build error
Trent
commited on
Commit
β’
532be59
1
Parent(s):
c0d4b59
Update model name
Browse files- app.py +1 -1
- embed.py +1 -1
- features/val2017/{koclip.tsv β koclip-base.tsv} +0 -0
app.py
CHANGED
@@ -7,7 +7,7 @@ import text2image
|
|
7 |
PAGES = {"Text to Image": text2image, "Image to Text": image2text}
|
8 |
|
9 |
st.sidebar.title("Navigation")
|
10 |
-
model = st.sidebar.selectbox("Choose a model", ["koclip", "koclip-large"])
|
11 |
page = st.sidebar.selectbox("Choose a task", list(PAGES.keys()))
|
12 |
|
13 |
PAGES[page].app(model)
|
|
|
7 |
PAGES = {"Text to Image": text2image, "Image to Text": image2text}
|
8 |
|
9 |
st.sidebar.title("Navigation")
|
10 |
+
model = st.sidebar.selectbox("Choose a model", ["koclip-base", "koclip-large"])
|
11 |
page = st.sidebar.selectbox("Choose a task", list(PAGES.keys()))
|
12 |
|
13 |
PAGES[page].app(model)
|
embed.py
CHANGED
@@ -16,7 +16,7 @@ def main(args):
|
|
16 |
files = list(os.listdir(root))
|
17 |
for f in files:
|
18 |
assert(f[-4:] == ".jpg")
|
19 |
-
for model_name in ["koclip", "koclip-large"]:
|
20 |
model, processor = load_model(f"koclip/{model_name}")
|
21 |
with tqdm(total=len(files)) as pbar:
|
22 |
for counter in range(0, len(files), args.batch_size):
|
|
|
16 |
files = list(os.listdir(root))
|
17 |
for f in files:
|
18 |
assert(f[-4:] == ".jpg")
|
19 |
+
for model_name in ["koclip-base", "koclip-large"]:
|
20 |
model, processor = load_model(f"koclip/{model_name}")
|
21 |
with tqdm(total=len(files)) as pbar:
|
22 |
for counter in range(0, len(files), args.batch_size):
|
features/val2017/{koclip.tsv β koclip-base.tsv}
RENAMED
File without changes
|