m7mdal7aj commited on
Commit
d26dd8d
1 Parent(s): 08ec8d2

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. my_model/KBVQA.py +1 -1
my_model/KBVQA.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
4
  from typing import Optional
5
- from my_model.image_captioning import ImageCaptioningModel
6
  from my_model.object_detection import ObjectDetector
7
 
8
 
 
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
4
  from typing import Optional
5
+ from my_model.captioner.image_captioning import ImageCaptioningModel
6
  from my_model.object_detection import ObjectDetector
7
 
8