Update app.py

app.py

@@ -399,7 +399,19 @@ def segment(image):
     plt.rcParams['ytick.left'] = False
     plt.rcParams['ytick.right'] = False
     plt.rcParams['figure.figsize'] = [15,10]
-    print(type(image))
-    pass  # Implement your image segmentation model here...
+    date_captured = Image.open(image)._getexif()[36867]
+    context_feature, score, results = run_inference(
+        faster_rcnn_model, image, date_captured)
+    faster_rcnn_output_dict = results
+    context_rcnn_image, faster_rcnn_image = show_context_rcnn_inference(
+        context_rcnn_model, image, context_features_matrix,
+        faster_rcnn_output_dict, context_padding_size)
+    plt.subplot(1,2,1)
+    plt.imshow(faster_rcnn_image)
+    plt.title('Faster R-CNN')
+    plt.subplot(1,2,2)
+    plt.imshow(context_rcnn_image)
+    plt.title('Context R-CNN')
+    return plt
 
 gr.Interface(fn=segment, inputs="image", outputs="image").launch()