Upload folder using huggingface_hub

app.py

@@ -1,5 +1,13 @@
-# import os
-# os.environ["HF_HOME"] = "/tmp/hf_cache"
+import os
+import shutil
+
+# Clean and recreate HF cache directory
+cache_dir = "/tmp/hf_cache"
+if os.path.exists(cache_dir):
+    shutil.rmtree(cache_dir)
+os.makedirs(cache_dir, exist_ok=True)
+os.environ["HF_HOME"] = cache_dir
+
 import streamlit as st
 from PIL import Image
 import torch
@@ -128,7 +136,6 @@ def apply_transform(image, size_mode=512):
 # Streamlit UI with Colorful Title and Emojis
 # st.markdown("<div style='text-align: center;'><span style='font-size: 3rem;'>🩺</span></div>", 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("<div style='text-align: center;'><span style='font-size: 3rem;'>🚀</span></div>", 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>", 
@@ -301,10 +308,7 @@ with col2:
     - The model’s responses may contain **<span style='color:#dc3545; font-weight:bold;'>🤖 hallucinations or incorrect information</span>**. 
     - Always consult a **<span style='color:#dc3545; font-weight:bold;'>👨‍⚕️ medical professional</span>** for accurate diagnosis and advice.
 """, unsafe_allow_html=True)
-st.markdown("""
-    <img src="//www.clustrmaps.com/map_v2.png?d=uM9v_RTadJ3hLvNbBSQ2PZ0KNPABbilkZgDyiXmuC0M&cl=ffffff" 
-         style="position:absolute;top:-9999px;left:-9999px;width:1px;height:1px;visibility:hidden;opacity:0;pointer-events:none;z-index:-1;display:none;" />
-""", unsafe_allow_html=True)
+
 # Run inference after upload
 if st.button("Run Inference 🏃‍♂️"):
     if image is None:

requirements.txt

@@ -1,6 +1,6 @@
 streamlit
 torch
-transformers
+transformers==4.46.3
 pillow
 numpy
 supervision