Update app.py

app.py

@@ -2,29 +2,24 @@
 import importlib
 import gradio as gr
 
-# Function to load the model dynamically
 def load_model(model_name):
     module = importlib.import_module(model_name)
     return module
 
-# Define the models
 models = {
     "Multi-class model": "model_1",
     "Empty class": "model_2",
     "Misalignment class": "model_3"
 }
 
-# Function to perform image detection
 def detect_image(model_choice, input_image):
     model = load_model(models[model_choice])
     return model.detect_image(input_image)
 
-# Function to perform video detection
 def detect_video(model_choice, input_video):
     model = load_model(models[model_choice])
     return model.detect_video(input_video)
 
-# Create Gradio app
 app = gr.Blocks()
 
 with app: