Update app.py

app.py

@@ -19,6 +19,21 @@ def image_to_base64_url(img):
     img_str = base64.b64encode(buffered.getvalue()).decode()
     return f"data:image/png;base64,{img_str}"
 
+
+# Authenticate with Hugging Face
+def authenticate():
+    token = st.text_input("Enter your Hugging Face token:", type="password", 
+                          help="Find your token at https://huggingface.co/settings/tokens")
+    if token:
+        try:
+            login(token)
+            st.success("Successfully authenticated with Hugging Face!")
+            return True
+        except Exception as e:
+            st.error(f"Authentication failed: {e}")
+            return False
+    return False
+    
 # Load model (wrapped in a function to load only when needed)
 @st.cache_resource
 def load_model():