shivangibithel commited on
Commit
cf9d9e0
·
1 Parent(s): f352d7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -1,20 +1,16 @@
1
  import streamlit as st
2
- st.set_page_config(page_title='Image Search App', layout='wide')
 
3
  import torch
4
  from transformers import AutoTokenizer, AutoModel
5
  import faiss
6
  import numpy as np
7
  import wget
8
  from PIL import Image
9
- # import io
10
  from sentence_transformers import SentenceTransformer
11
  import json
12
  from zipfile import ZipFile
13
  import zipfile
14
- # from io import BytesIO
15
- from PIL import Image
16
- # from huggingface_hub import hf_hub_download
17
- # hf_hub_download(repo_id="shivangibithel/Flickr8k", filename="Images.zip")
18
 
19
  # Load the pre-trained sentence encoder
20
  model_name = "sentence-transformers/all-distilroberta-v1"
@@ -61,9 +57,6 @@ def search(query, k=5):
61
  image = Image.open(image_data)
62
  st.image(image, width=600)
63
 
64
-
65
- st.title("Image Search App")
66
-
67
  query = st.text_input("Enter your search query here:")
68
  if st.button("Search"):
69
  if query:
 
1
  import streamlit as st
2
+ st.set_page_config(page_title='T2I', page_icon="🧊", layout='centered')
3
+ st.title("Text To Image Retrieval for KaggleX BPIOC Mentorship Program")
4
  import torch
5
  from transformers import AutoTokenizer, AutoModel
6
  import faiss
7
  import numpy as np
8
  import wget
9
  from PIL import Image
 
10
  from sentence_transformers import SentenceTransformer
11
  import json
12
  from zipfile import ZipFile
13
  import zipfile
 
 
 
 
14
 
15
  # Load the pre-trained sentence encoder
16
  model_name = "sentence-transformers/all-distilroberta-v1"
 
57
  image = Image.open(image_data)
58
  st.image(image, width=600)
59
 
 
 
 
60
  query = st.text_input("Enter your search query here:")
61
  if st.button("Search"):
62
  if query: