Update app.py

app.py

@@ -3,6 +3,7 @@ from datasets import load_dataset
 import random
 import numpy as np
 from transformers import CLIPProcessor, CLIPModel
+import os.environ
 
 # Load the pre-trained model and processor
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
@@ -28,7 +29,7 @@ def rand_image():
 def set_labels(text):
     return text.split(",")
     
-get_caption = gr.load("ryaalbr/caption", src="spaces")
+get_caption = gr.load("ryaalbr/caption", src="spaces", hf_token=os.environ("api_key"))
 def generate_text(image):
     return get_caption(image)