Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -13,7 +13,7 @@ pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-jaguar", torch_dtype | |
| 13 | 
             
            MAX_SEED = np.iinfo(np.int32).max
         | 
| 14 | 
             
            MAX_IMAGE_SIZE = 2048
         | 
| 15 |  | 
| 16 | 
            -
            @spaces.GPU(duration= | 
| 17 | 
             
            def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
         | 
| 18 | 
             
                """ 
         | 
| 19 | 
             
                Generate an image using the Shuttle Jaguar Image Generator.
         | 
| @@ -101,7 +101,7 @@ Shuttle Jaguar is a text-to-image AI model designed to generate highly aesthetic | |
| 101 | 
             
                            num_inference_steps = gr.Slider(
         | 
| 102 | 
             
                                label="Number of inference steps",
         | 
| 103 | 
             
                                minimum=1,
         | 
| 104 | 
            -
                                maximum= | 
| 105 | 
             
                                step=1,
         | 
| 106 | 
             
                                value=4,
         | 
| 107 | 
             
                            )
         | 
|  | |
| 13 | 
             
            MAX_SEED = np.iinfo(np.int32).max
         | 
| 14 | 
             
            MAX_IMAGE_SIZE = 2048
         | 
| 15 |  | 
| 16 | 
            +
            @spaces.GPU(duration=10)
         | 
| 17 | 
             
            def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
         | 
| 18 | 
             
                """ 
         | 
| 19 | 
             
                Generate an image using the Shuttle Jaguar Image Generator.
         | 
|  | |
| 101 | 
             
                            num_inference_steps = gr.Slider(
         | 
| 102 | 
             
                                label="Number of inference steps",
         | 
| 103 | 
             
                                minimum=1,
         | 
| 104 | 
            +
                                maximum=16,
         | 
| 105 | 
             
                                step=1,
         | 
| 106 | 
             
                                value=4,
         | 
| 107 | 
             
                            )
         | 
