Spaces:
				
			
			
	
			
			
		Sleeping
		
	
	
	
			
			
	
	
	
	
		
		
		Sleeping
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -23,7 +23,7 @@ def load_model(): | |
| 23 | 
             
                    )
         | 
| 24 |  | 
| 25 | 
             
                # Load the processor and model using the correct identifier
         | 
| 26 | 
            -
                model_id = "google/paligemma2-10b-pt- | 
| 27 | 
             
                processor = PaliGemmaProcessor.from_pretrained(model_id, use_auth_token=token)
         | 
| 28 | 
             
                device = "cuda" if torch.cuda.is_available() else "cpu"
         | 
| 29 | 
             
                model = PaliGemmaForConditionalGeneration.from_pretrained(
         | 
|  | |
| 23 | 
             
                    )
         | 
| 24 |  | 
| 25 | 
             
                # Load the processor and model using the correct identifier
         | 
| 26 | 
            +
                model_id = "google/paligemma2-10b-pt-448"
         | 
| 27 | 
             
                processor = PaliGemmaProcessor.from_pretrained(model_id, use_auth_token=token)
         | 
| 28 | 
             
                device = "cuda" if torch.cuda.is_available() else "cpu"
         | 
| 29 | 
             
                model = PaliGemmaForConditionalGeneration.from_pretrained(
         |