From 16b735b16240e116bf5c6e2e9c852181d91c1554 Mon Sep 17 00:00:00 2001
From: root <“junnan.li@salesforce.com”>
Date: Fri, 28 Jan 2022 13:36:42 +0000
Subject: [PATCH] update demo

---
 demo.ipynb | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/demo.ipynb b/demo.ipynb
index b86316a..92647d9 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "bb2e6f60",
+   "id": "2b949f9f",
    "metadata": {},
    "source": [
     "# BLIP: Inference Demo"
@@ -93,7 +93,7 @@
     "\n",
     "model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_base_caption.pth'\n",
     "    \n",
-    "model = blip_decoder(pretrained=model_url, image_size=384, vit='base')\n",
+    "model = blip_decoder(pretrained=model_url, image_size=384, vit='base', med_config='BLIP/configs/med_config.json')\n",
     "model.eval()\n",
     "model = model.to(device)\n",
     "\n",
@@ -139,7 +139,7 @@
     "\n",
     "model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_vqa.pth'\n",
     "    \n",
-    "model = blip_vqa(pretrained=model_url, image_size=480, vit='base')\n",
+    "model = blip_vqa(pretrained=model_url, image_size=480, vit='base', med_config='BLIP/configs/med_config.json')\n",
     "model.eval()\n",
     "model = model.to(device)\n",
     "\n",