Upload app.py with huggingface_hub

app.py

@@ -51,11 +51,13 @@ def load_knn():
     return np.load(knnpath)
 
 def main():
-    st.title("Percept Human Perception Analyzer")
+    st.title("Percept: Human Perception of Street View Image Analyzer")
  
     try:
         with st.spinner('Loading CLIP model... This may take a moment.'):
             model, preprocess, tokenizer = load_model()
+        device = "cuda" if torch.cuda.is_available() else "cpu"
+        model = model.to(device)
     except Exception as e:
         st.error(f"Error loading model: {str(e)}")
         st.info("Please make sure you have enough memory and the correct dependencies installed.")
@@ -68,9 +70,22 @@ def main():
 
     if file:
         try:
-            with Image.open(file) as img:
-                st.write(file)
-                st.write(img.size)
+            image = Image.open(file)
+
+            st.image(image, caption="Uploaded Image", use_column_width=True)
+                
+            # Process image
+            with st.spinner('Processing image...'):
+                processed_image = process_image(image, preprocess)
+                processed_image = processed_image.to(device)
+
+                # Encode into CLIP vector
+                with torch.no_grad():
+                    vec = model.encode_image(processed_image)
+                    
+                    # Normalize vector
+                    vec /= vec.norm(dim=-1, keepdim=True)
+                st.write(vec.shape)
         except Exception as e:
             st.error(f"Error processing image: {str(e)}")