Upload folder using huggingface_hub

fal_image_generator.py

@@ -5,8 +5,9 @@ import requests
 from io import BytesIO
 
 from weave_prompt import TextToImageModel
-import load_keys
 
+from dotenv import load_dotenv
+load_dotenv()
 
 class FalImageGenerator(TextToImageModel):
     """Handles image generation using fal_client."""

prompt_refiner.py

@@ -5,7 +5,6 @@ import os
 from dotenv import load_dotenv
 
 from weave_prompt import PromptRefiner
-import load_keys
 
 # Load environment variables from .env file
 load_dotenv()