Spaces:
Sleeping
Sleeping
import os | |
import sys | |
import gradio as gr | |
from utils import load_embeddings | |
from ui import categorize_products, create_demo # Updated imports | |
# Path to the embeddings file | |
EMBEDDINGS_PATH = "ingredient_embeddings_voyageai.pkl" | |
# Check if embeddings file exists | |
if not os.path.exists(EMBEDDINGS_PATH): | |
print(f"Error: Embeddings file {EMBEDDINGS_PATH} not found!") | |
print(f"Please ensure the file exists at {os.path.abspath(EMBEDDINGS_PATH)}") | |
sys.exit(1) | |
# Load embeddings globally | |
try: | |
embeddings_data = load_embeddings(EMBEDDINGS_PATH) | |
# Make embeddings available to the UI functions | |
import ui | |
ui.embeddings = embeddings_data | |
except Exception as e: | |
print(f"Error loading embeddings: {e}") | |
sys.exit(1) | |
# Launch the Gradio interface | |
if __name__ == "__main__": | |
demo = create_demo() | |
demo.launch() |