Upload folder using huggingface_hub

app.py

@@ -1,4 +1,6 @@
 
+import os
+os.environ["HF_HOME"] = "/tmp/hf_cache"
 import streamlit as st
 from PIL import Image
 import torch
@@ -13,68 +15,6 @@ import yaml
 # Set Streamlit page configuration for a wide layout
 st.set_page_config(layout="wide")
 
-# Custom CSS for better layout and mobile responsiveness
-# st.markdown("""
-#     <style>
-#         .main {
-#             max-width: 1200px;  /* Max width for content */
-#             margin: 0 auto;
-#         }
-#         .block-container {
-#             padding-top: 2rem;
-#             padding-bottom: 2rem;
-#             padding-left: 3rem;
-#             padding-right: 3rem;
-#         }
-#         .title {
-#             font-size: 3.2rem;
-#             text-align: center;
-#             background: linear-gradient(135deg, #0575e6 0%, #ff0080 50%, #7928ca 100%);
-#             -webkit-background-clip: text;
-#             -webkit-text-fill-color: transparent;
-#             background-clip: text;
-#         }
-        
-#         @keyframes gradientShift {
-#             0% { background-position: 0% 50%; }
-#             50% { background-position: 100% 50%; }
-#             100% { background-position: 0% 50%; }
-#         }
-#         .subheader {
-#             font-size: 1.5rem;
-#             margin-bottom: 20px;
-#         }
-#         .btn {
-#             font-size: 1.1rem;
-#             padding: 10px 20px;
-#             background-color: #FF6347;
-#             color: white;
-#             border-radius: 5px;
-#             border: none;
-#             cursor: pointer;
-#         }
-#         .btn:hover {
-#             background-color: #FF4500;
-#         }
-#         .column-spacing {
-#             display: flex;
-#             justify-content: space-between;
-#         }
-#         .col-half {
-#             width: 48%;
-#         }
-#         .col-full {
-#             width: 100%;
-#         }
-#         .instructions {
-#             padding: 20px;
-#             background-color: #f9f9f9;
-#             border-radius: 8px;
-#             box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
-#         }
-#     </style>
-# """, unsafe_allow_html=True)
-
 # Custom CSS for better layout and mobile responsiveness
 st.markdown("""
     <style>
@@ -89,9 +29,18 @@ st.markdown("""
             padding-right: 3rem;
         }
         .title {
-            font-size: 2.5rem;
+            font-size: 3.2rem;
             text-align: center;
-            color: #FF6347;
+            background: linear-gradient(135deg, #0575e6 0%, #ff0080 50%, #7928ca 100%);
+            -webkit-background-clip: text;
+            -webkit-text-fill-color: transparent;
+            background-clip: text;
+        }
+        
+        @keyframes gradientShift {
+            0% { background-position: 0% 50%; }
+            50% { background-position: 100% 50%; }
+            100% { background-position: 0% 50%; }
         }
         .subheader {
             font-size: 1.5rem;
@@ -128,6 +77,59 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
+# # Custom CSS for better layout and mobile responsiveness
+# st.markdown("""
+#     <style>
+#         .main {
+#             max-width: 1200px;  /* Max width for content */
+#             margin: 0 auto;
+#         }
+#         .block-container {
+#             padding-top: 2rem;
+#             padding-bottom: 2rem;
+#             padding-left: 3rem;
+#             padding-right: 3rem;
+#         }
+#         .title {
+#             font-size: 2.5rem;
+#             text-align: center;
+#             color: #FF6347;
+#         }
+#         .subheader {
+#             font-size: 1.5rem;
+#             margin-bottom: 20px;
+#         }
+#         .btn {
+#             font-size: 1.1rem;
+#             padding: 10px 20px;
+#             background-color: #FF6347;
+#             color: white;
+#             border-radius: 5px;
+#             border: none;
+#             cursor: pointer;
+#         }
+#         .btn:hover {
+#             background-color: #FF4500;
+#         }
+#         .column-spacing {
+#             display: flex;
+#             justify-content: space-between;
+#         }
+#         .col-half {
+#             width: 48%;
+#         }
+#         .col-full {
+#             width: 100%;
+#         }
+#         .instructions {
+#             padding: 20px;
+#             background-color: #f9f9f9;
+#             border-radius: 8px;
+#             box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
+#         }
+#     </style>
+# """, unsafe_allow_html=True)
+
 # Load Model and Processor
 @st.cache_resource
 def load_model():
@@ -186,8 +188,8 @@ def apply_transform(image, size_mode=512):
     return transformed["image"]
 
 # Streamlit UI with Colorful Title and Emojis
+st.markdown("<h1 class='title'>Knowledge to Sight: Reasoning over Visual Attributes via Knowledge Decomposition for Abnormality Grounding </h1>", unsafe_allow_html=True)
 # st.markdown("<h1 class='title'>🩺 Knowledge to Sight: Reasoning over Visual Attributes via Knowledge Decomposition for Abnormality Grounding 🚀</h1>", unsafe_allow_html=True)
-st.markdown("<h1 class='title'>🩺 Knowledge to Sight: Reasoning over Visual Attributes via Knowledge Decomposition for Abnormality Grounding 🚀</h1>", unsafe_allow_html=True)
 st.markdown(
     "<p style='text-align: center; font-size: 18px;'>Welcome to a simple demo of our work! 🎉 Choose an example or upload your own image to get started! 👇</p>", 
     unsafe_allow_html=True