Update app.py

app.py

@@ -1,26 +1,31 @@
+# app.py
 import gradio as gr
 import importlib
 
-def launch_model_app(model_script):
-    model_module = importlib.import_module(model_script)
-    model_app = model_module.app
-    model_app.launch()
+def load_model(model_name):
+    module = importlib.import_module(model_name)
+    return module
 
-app = gr.Interface(
-    launch_model_app,
-    inputs=[
-        gr.inputs.Dropdown(
-            choices=[
-                ("Model 1", "app_model_1"),
-                ("Model 2", "app_model_2"),
-                ("Model 3", "app_model_3")
-            ],
-            label="Select Model"
-        )
-    ],
-    outputs=None,
-    title="Choose Object Detection Model",
-    theme="compact"
-)
+app = gr.Blocks()
 
-app.launch()
+with app:
+    gr.Markdown("## Object Detection using TensorFlow Lite Models")
+    with gr.Row():
+        model_choice = gr.Dropdown(label="Select Model", choices=["model_1", "model_2", "model_3"])
+        image_input = gr.Image(type="pil", label="Upload an image")
+        image_output = gr.Image(type="pil", label="Detection Result")
+        video_input = gr.Video(label="Upload a video")
+        video_output = gr.Video(label="Detection Result")
+        
+        def image_detection(model_name, input_image):
+            model = load_model(model_name)
+            return model.detect_image(input_image)
+        
+        def video_detection(model_name, input_video):
+            model = load_model(model_name)
+            return model.detect_video(input_video)
+        
+        gr.Button("Submit Image").click(fn=image_detection, inputs=[model_choice, image_input], outputs=image_output)
+        gr.Button("Submit Video").click(fn=video_detection, inputs=[model_choice, video_input], outputs=video_output)
+
+app.launch()