Upload folder using huggingface_hub

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from PIL import Image
 import time
+from dotenv import load_dotenv
 from image_to_text import LlamaEvaluator
 from prompt_refiner import LlamaPromptRefiner
 from weave_prompt import PromptOptimizer
@@ -9,6 +10,9 @@ from lpips_evaluator import LPIPSImageSimilarityMetric
 from fal_image_generator import FalImageGenerator
 import io
 
+# Load environment variables from .env file
+load_dotenv()
+
 st.set_page_config(
     page_title="WeavePrompt Demo",
     page_icon="🎨",

image_to_text.py

@@ -8,7 +8,10 @@ from pathlib import Path
 from PIL import Image
 from typing import Dict, Any, Optional
 from weave_prompt import ImageEvaluator
-import load_keys
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 
 # Weave autopatches OpenAI to log LLM calls to W&B
 weave.init("meta-llama")

prompt_refiner.py

@@ -2,10 +2,14 @@ from typing import Any, Dict
 import openai
 import weave
 import os
+from dotenv import load_dotenv
 
 from weave_prompt import PromptRefiner
 import load_keys
 
+# Load environment variables from .env file
+load_dotenv()
+
 # Weave autopatches OpenAI to log LLM calls to W&B
 weave.init(project_name="meta-llama")