updates

app.py

@@ -3,8 +3,12 @@ import os
 import threading
 import random
 from datasets import load_dataset, Dataset, Features, Value
-
-# Set up your Hugging Face dataset name
+from huggingface_hub import login
+token = os.getenv("HUGGINGFACE_TOKEN")
+if token:
+    login(token=token)
+else:
+    print("HUGGINGFACE_TOKEN environment variable not set.")
 dataset_name = "GeorgeIbrahim/EGYCOCO"  # Replace with your dataset name
 
 # Load or create the dataset
@@ -108,4 +112,4 @@ with gr.Blocks() as demo:
     # Load initial image
     demo.load(fn=initialize_interface, inputs=None, outputs=[image, caption])
 
-demo.launch(share=True)
+demo.launch(share=True)