abreza's picture
refactor: replace load_model with get_model for improved model loading
ea52793
raw
history blame contribute delete
493 Bytes
from interface import create_interface
from model_utils import get_model
def main():
print("Loading model...")
model = get_model()
if model is None:
print("Warning: Model could not be loaded. Please ensure 'best_model.pkl' exists.")
else:
print("Model loaded successfully!")
print("Creating interface...")
demo = create_interface()
print("Launching application...")
demo.launch(share=True)
if __name__ == "__main__":
main()