Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -48,7 +48,7 @@ with gr.Blocks() as demo: | |
| 48 | 
             
                def predict(img, model_name):
         | 
| 49 | 
             
                    if model_name == "CLIP-GPT2":
         | 
| 50 | 
             
                        return generate_caption_clipgpt(img)  
         | 
| 51 | 
            -
                    elif model_name == "ViT-GPT2"
         | 
| 52 | 
             
                        return generate_caption_vitgpt(img)
         | 
| 53 | 
             
                    else:
         | 
| 54 | 
             
                        return "Caption generation for this model is not yet implemented."
         | 
|  | |
| 48 | 
             
                def predict(img, model_name):
         | 
| 49 | 
             
                    if model_name == "CLIP-GPT2":
         | 
| 50 | 
             
                        return generate_caption_clipgpt(img)  
         | 
| 51 | 
            +
                    elif model_name == "ViT-GPT2":
         | 
| 52 | 
             
                        return generate_caption_vitgpt(img)
         | 
| 53 | 
             
                    else:
         | 
| 54 | 
             
                        return "Caption generation for this model is not yet implemented."
         |