Update app.py

app.py

@@ -35,6 +35,13 @@ import streamlit as st
 from data import flight_data
 
 
+
+hf_token = os.getenv("HF_TOKEN")
+if hf_token is None:
+    raise ValueError("Hugging Face token not found. Please set the HF_TOKEN environment variable.")
+login(hf_token)
+
+
 API_URL = "https://api-inference.huggingface.co/models/google/tapas-base-finetuned-wtq"
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}