Update app.py

app.py

@@ -18,6 +18,13 @@ from BLIP.models.blip import blip_decoder
 
 import gradio as gr
 
+device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+blip_image_eval_size = 384
+blip_model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_base_caption.pth'        
+blip_model = blip_decoder(pretrained=blip_model_url, image_size=blip_image_eval_size, vit='base')
+blip_model.eval()
+blip_model = blip_model.to(device)
+
 def greet(name):
     return "hi " + name + "!!"