Update app.py

app.py

@@ -11,6 +11,11 @@ import gradio as gr
 image_folder = "Images/"
 metadata_file = "descriptions.json"
 
+# Define the function to load metadata
+def load_metadata(metadata_file):
+    with open(metadata_file, 'r') as f:
+        metadata = json.load(f)
+    return metadata
 
 # Custom Dataset Class
 class ImageDescriptionDataset(Dataset):
@@ -83,14 +88,15 @@ def train_lora(image_folder, metadata):
 
     print("Training completed.")
 
-# Define Gradio app
-def start_training():
+# Gradio app function to load metadata and start training
+def start_training_gradio():
     print("Preparing dataset...")
+    metadata = load_metadata(metadata_file)  # Load metadata
     return train_lora(image_folder, metadata)
 
 # Gradio interface
 demo = gr.Interface(
-    fn=start_training,
+    fn=start_training_gradio,  # Use the new function name here
     inputs=None,
     outputs="text",
     title="Train LoRA on Your Dataset",