File size: 1,244 Bytes
5e6614b b0d8677 5e6614b b0d8677 5e6614b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import gradio as gr
import json
import os
# Paths
image_folder = "Images/" # Folder containing the images
metadata_file = "descriptions.json" # JSON file with image descriptions
# Load metadata
with open(metadata_file, "r") as f:
metadata = json.load(f)
# Placeholder function for training LoRA
def train_lora(image_folder, metadata):
# Prepare a dataset of image paths and descriptions
dataset = []
for image_name, description in metadata.items():
image_path = os.path.join(image_folder, image_name)
if os.path.exists(image_path): # Ensure the image file exists
dataset.append({"image": image_path, "description": description})
else:
print(f"Warning: {image_name} not found in {image_folder}")
# Placeholder for training logic
num_images = len(dataset)
return f"Training LoRA with {num_images} images and their descriptions."
# Define Gradio app
def start_training():
return train_lora(image_folder, metadata)
# Gradio interface
demo = gr.Interface(
fn=start_training,
inputs=None,
outputs="text",
title="Train LoRA on Your Dataset",
description="Click below to start training with the uploaded images and metadata."
)
demo.launch()
|