Update app.py

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image
 from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
-from display import inference
+from display import inference, draw_predictions
 
 
 gr.Interface(
@@ -18,7 +18,7 @@ gr.Interface(
         gr.Slider(0, 1, value=0.5, label="Opacity of GradCAM"),        
     ],
     outputs=gr.Gallery(rows=2, columns=1),
-    title = "Object Detection : YoloV3 on PASCAL VOC Dataset From Scratch" 
+    title = "Object Detection : YoloV3 on PASCAL VOC Dataset From Scratch (with GradCAM)" 
     
     ,examples=[
        ["Examples/000001.jpg", 0.75, 0.75, True, 0.5],