Update app.py

app.py

@@ -17,7 +17,7 @@ import matplotlib.patches as patches
 
 # Load the model
 model = YoloVersion3( )
-model.load_state_dict(torch.load('/content/drive/MyDrive/sunandini/Checkpoint/lightning_logs/version_4/checkpoints/Yolov3.pth', map_location=torch.device('cpu')), strict=False)
+model.load_state_dict(torch.load('Yolov3.pth', map_location=torch.device('cpu')), strict=False)
 model.eval()
 
 # Anchor
@@ -136,7 +136,7 @@ def inference(inp_image):
 inputs = gr.inputs.Image(label="Original Image")
 outputs = gr.outputs.Image(type="pil",label="Output Image")
 title = "YOLOv3 model trained on PASCAL VOC Dataset"
-description = "YOLOv3 Gradio demo for object detection"
-examples = [['/content/car1.jpg'], ['/content/home.jpg']]
-gr.Interface(inference, inputs, outputs, title=title,  examples=examples, description=description, theme='abidlabs/dracula_revamped').launch(
+description = "YOLOv3 object detection using Gradio demo"
+examples = [['car.jpg'], ['home.jpg'],['train.jpg'],['train_persons.jpg']]
+gr.Interface(inference, inputs, outputs, title=title,  examples=examples, description=description, theme='xiaobaiyuan/theme_brief').launch(
     debug=False)