Update app.py

app.py

@@ -15,7 +15,7 @@ def detect(inp):
   
 
 
- def custom(path_or_model='path/to/model.pt', autoshape=True):
+def custom(path_or_model='path/to/model.pt', autoshape=True):
     """custom mode
     Arguments (3 options):
         path_or_model (str): 'path/to/model.pt'
@@ -27,7 +27,7 @@ def detect(inp):
     model = torch.load(path_or_model) if isinstance(path_or_model, str) else path_or_model  # load checkpoint
     if isinstance(model, dict):
         model = model['ema' if model.get('ema') else 'model']  # load model
-
+    
     hub_model = Model(model.yaml).to(next(model.parameters()).device)  # create
     hub_model.load_state_dict(model.float().state_dict())  # load state_dict
     hub_model.names = model.names  # class names
@@ -35,7 +35,7 @@ def detect(inp):
         hub_model = hub_model.autoshape()  # for file/URI/PIL/cv2/np inputs and NMS
     device = select_device('0' if torch.cuda.is_available() else 'cpu')  # default to GPU if available
     return hub_model.to(device)
-
+    
 model = custom(path_or_model='best.pt')
 
 def detect1(inp):