KB-VQA-E

Running

m7mdal7aj commited on Feb 23

Commit

d26dd8d

•

1 Parent(s): 08ec8d2

Update my_model/KBVQA.py

Files changed (1) hide show

my_model/KBVQA.py CHANGED Viewed

@@ -2,7 +2,7 @@ import streamlit as st
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import Optional
-from my_model.image_captioning import ImageCaptioningModel
 from my_model.object_detection import ObjectDetector

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import Optional
+from my_model.captioner.image_captioning import ImageCaptioningModel
 from my_model.object_detection import ObjectDetector