import gradio as gr
from src.models.text_classification import TextClassificationModel
import logging

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Initialize model
logger.info("Initializing Text Classification model...")
model = TextClassificationModel()

# Create the interface
logger.info("Creating Gradio interface...")
demo = model.create_interface()

# Launch the interface
demo.launch()