Upload folder using huggingface_hub

app.py

@@ -1,20 +1,17 @@
 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
-from mock_components import MockTextToImageModel, MockImageEvaluator, MockPromptRefiner
 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_title="WeavePrompt",
     page_icon="🎨",
     layout="wide"
 )
@@ -23,7 +20,6 @@ def main():
     st.title("🎨 WeavePrompt: Iterative Prompt Optimization")
     st.markdown("""
     Upload a target image and watch as WeavePrompt iteratively optimizes a text prompt to recreate it.
-    This demo uses mock components for illustration.
     """)
     
     # Initialize session state