Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -36,7 +36,8 @@ if __name__ == "__main__": | |
| 36 | 
             
                # def 
         | 
| 37 | 
             
                model = instantiate(cfg.model)
         | 
| 38 | 
             
                checkpointer = DetectionCheckpointer(model)
         | 
| 39 | 
            -
                checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_voc.pth")
         | 
|  | |
| 40 |  | 
| 41 | 
             
                model.eval()
         | 
| 42 | 
             
                model.cuda()
         | 
|  | |
| 36 | 
             
                # def 
         | 
| 37 | 
             
                model = instantiate(cfg.model)
         | 
| 38 | 
             
                checkpointer = DetectionCheckpointer(model)
         | 
| 39 | 
            +
                # checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_voc.pth")
         | 
| 40 | 
            +
                checkpointer.load("https://huggingface.co/allencbzhang/vCLR/resolve/main/vCLR_deformable_train_on_coco.pth")
         | 
| 41 |  | 
| 42 | 
             
                model.eval()
         | 
| 43 | 
             
                model.cuda()
         |