Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	Commit 
							
							·
						
						c7580e7
	
1
								Parent(s):
							
							a97fda4
								
Removed extra spaces.GPU() decorator.
Browse files
    	
        app.py
    CHANGED
    
    | 
         @@ -55,12 +55,12 @@ def load_yolo_model(model_name, confidence_threshold=0.5): 
     | 
|
| 55 | 
         
             
                global model
         
     | 
| 56 | 
         
             
                model_path = model_name
         
     | 
| 57 | 
         
             
                model = AutoDetectionModel.from_pretrained(
         
     | 
| 58 | 
         
            -
                    model_type="ultralytics", model_path=model_path, device= 
     | 
| 59 | 
         
             
                    confidence_threshold=confidence_threshold, image_size=IMAGE_SIZE
         
     | 
| 60 | 
         
             
                )
         
     | 
| 61 | 
         
             
                return model
         
     | 
| 62 | 
         | 
| 63 | 
         
            -
            @spaces.GPU()
         
     | 
| 64 | 
         
             
            def sahi_yolo_inference(
         
     | 
| 65 | 
         
             
                image,
         
     | 
| 66 | 
         
             
                yolo_model_name,
         
     | 
| 
         | 
|
| 55 | 
         
             
                global model
         
     | 
| 56 | 
         
             
                model_path = model_name
         
     | 
| 57 | 
         
             
                model = AutoDetectionModel.from_pretrained(
         
     | 
| 58 | 
         
            +
                    model_type="ultralytics", model_path=model_path, device=None, # auto device selection
         
     | 
| 59 | 
         
             
                    confidence_threshold=confidence_threshold, image_size=IMAGE_SIZE
         
     | 
| 60 | 
         
             
                )
         
     | 
| 61 | 
         
             
                return model
         
     | 
| 62 | 
         | 
| 63 | 
         
            +
            @spaces.GPU(duration=60)
         
     | 
| 64 | 
         
             
            def sahi_yolo_inference(
         
     | 
| 65 | 
         
             
                image,
         
     | 
| 66 | 
         
             
                yolo_model_name,
         
     |