Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
|
|
8 |
import torch.nn as nn
|
9 |
from my_model.object_detection import detect_and_draw_objects
|
10 |
from my_model.captioner.image_captioning import get_caption
|
11 |
-
from my_model.
|
12 |
from my_model.KBVQA import KBVQA, prepare_kbvqa_model
|
13 |
import my_model.utilities.st_config as st_config
|
14 |
|
|
|
8 |
import torch.nn as nn
|
9 |
from my_model.object_detection import detect_and_draw_objects
|
10 |
from my_model.captioner.image_captioning import get_caption
|
11 |
+
from my_model.gen_utilities import free_gpu_resources
|
12 |
from my_model.KBVQA import KBVQA, prepare_kbvqa_model
|
13 |
import my_model.utilities.st_config as st_config
|
14 |
|