push changes to hugging face

app.py

@@ -14,9 +14,11 @@ import os
 import matplotlib.pyplot as plt
 from utils import *
 
+checkpoint_filename = "object_detection_app/custom_yolo_v3.pt"
+relative_path = os.path.join(os.getcwd(),"hugging_space_app", checkpoint_filename)
 
 model = YOLOv3(num_classes=config.NUM_CLASSES).to(config.DEVICE)
-model.load_state_dict(torch.load("custom_yolo_v3.pt", map_location=torch.device('cpu')), strict=False)
+model.load_state_dict(torch.load(relative_path, map_location=torch.device('cpu')), strict=False)
 
 IMAGE_SIZE = config.IMAGE_SIZE
 test_transforms = A.Compose(
@@ -78,7 +80,7 @@ demo = gr.Interface(
               gr.Slider(0, 1, value = 0.75, label="Threshold"), 
               gr.Slider(0, 1, value = 0.75, label="IoU Threshold"), 
               gr.Slider(0, 1, value = 0.8, label="Opacity of GradCAM")],
-    outputs = [gr.Image(label="Output").scale(width=600, height=600)],
+    outputs = [gr.Image(width=600, height=600,label="Output")],
     title = title,
     description = description,
     examples = examples,