Update flask_app.py

flask_app.py

@@ -147,9 +147,10 @@ def img_text():
     
     sentences = data['sentences']
     print(sentences)
+    device = torch.device('cpu')
     
     image_size = 384
-    image = load_demo_image(image_size=image_size, img_url = image_url)
+    image = load_demo_image(image_size=image_size, device=device, img_url = image_url)
 
     
     model = blip_itm(pretrained=model_url, image_size=image_size, vit='base')