add full file upload & knn get score functionality
Browse files
app.py
CHANGED
|
@@ -76,7 +76,7 @@ def load_knn():
|
|
| 76 |
|
| 77 |
def main():
|
| 78 |
st.title("Percept: Human Perception of Street View Image Analyzer")
|
| 79 |
-
|
| 80 |
try:
|
| 81 |
with st.spinner('Loading CLIP model... This may take a moment.'):
|
| 82 |
model, preprocess, tokenizer = load_model()
|
|
@@ -96,8 +96,8 @@ def main():
|
|
| 96 |
try:
|
| 97 |
image = Image.open(file)
|
| 98 |
|
| 99 |
-
st.image(image, caption="Uploaded Image", width=
|
| 100 |
-
|
| 101 |
# Process image
|
| 102 |
with st.spinner('Processing image...'):
|
| 103 |
processed_image = process_image(image, preprocess)
|
|
@@ -106,7 +106,7 @@ def main():
|
|
| 106 |
# Encode into CLIP vector
|
| 107 |
with torch.no_grad():
|
| 108 |
vec = model.encode_image(processed_image)
|
| 109 |
-
|
| 110 |
# Normalize vector
|
| 111 |
vec /= vec.norm(dim=-1, keepdim=True)
|
| 112 |
if debug: st.write(vec.shape)
|
|
|
|
| 76 |
|
| 77 |
def main():
|
| 78 |
st.title("Percept: Human Perception of Street View Image Analyzer")
|
| 79 |
+
|
| 80 |
try:
|
| 81 |
with st.spinner('Loading CLIP model... This may take a moment.'):
|
| 82 |
model, preprocess, tokenizer = load_model()
|
|
|
|
| 96 |
try:
|
| 97 |
image = Image.open(file)
|
| 98 |
|
| 99 |
+
st.image(image, caption="Uploaded Image", width=400)
|
| 100 |
+
|
| 101 |
# Process image
|
| 102 |
with st.spinner('Processing image...'):
|
| 103 |
processed_image = process_image(image, preprocess)
|
|
|
|
| 106 |
# Encode into CLIP vector
|
| 107 |
with torch.no_grad():
|
| 108 |
vec = model.encode_image(processed_image)
|
| 109 |
+
|
| 110 |
# Normalize vector
|
| 111 |
vec /= vec.norm(dim=-1, keepdim=True)
|
| 112 |
if debug: st.write(vec.shape)
|