Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	track tqdm in Geadio UI
Browse files- gradio_app.py +1 -1
 
    	
        gradio_app.py
    CHANGED
    
    | 
         @@ -39,7 +39,7 @@ hf_hub_download( 
     | 
|
| 39 | 
         
             
            )
         
     | 
| 40 | 
         | 
| 41 | 
         
             
            @spaces.GPU()
         
     | 
| 42 | 
         
            -
            def generate(control_sequence, ref_image):
         
     | 
| 43 | 
         
             
                control_image = control_sequence # "data_test/sample4.mp4"
         
     | 
| 44 | 
         
             
                ref_image = ref_image # "data_test/sample4.png"
         
     | 
| 45 | 
         
             
                unique_id = str(uuid.uuid4())
         
     | 
| 
         | 
|
| 39 | 
         
             
            )
         
     | 
| 40 | 
         | 
| 41 | 
         
             
            @spaces.GPU()
         
     | 
| 42 | 
         
            +
            def generate(control_sequence, ref_image, progress=gr.Progress(track_tqdm=True)):
         
     | 
| 43 | 
         
             
                control_image = control_sequence # "data_test/sample4.mp4"
         
     | 
| 44 | 
         
             
                ref_image = ref_image # "data_test/sample4.png"
         
     | 
| 45 | 
         
             
                unique_id = str(uuid.uuid4())
         
     |