Ahmed-El-Sharkawy commited on
Commit
8eece3e
·
verified ·
1 Parent(s): b7a50ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -88,5 +88,22 @@ video_interface = gr.Interface(
88
  title="Video Inference"
89
  )
90
 
91
- demo = gr.TabbedInterface([image_interface, video_interface], ["Image", "Video"])
92
- demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
88
  title="Video Inference"
89
  )
90
 
91
+ with gr.Blocks() as demo:
92
+ with gr.TabItem("Image"):
93
+ gr.Interface(
94
+ fn=lambda img, model_name: predict_image(img, resnet_model if model_name == "ResNet50" else mobilenet_model),
95
+ inputs=inputs_image,
96
+ outputs=outputs_image,
97
+ title="Image Inference"
98
+ ).render()
99
+
100
+ with gr.TabItem("Video"):
101
+ gr.Interface(
102
+ fn=lambda vid, model_name: predict_video(vid, resnet_model if model_name == "ResNet50" else mobilenet_model),
103
+ inputs=inputs_video,
104
+ outputs=outputs_video,
105
+ title="Video Inference"
106
+ ).render()
107
+
108
+ demo.launch()
109
+