Update my_model/tabs/run_inference.py
Browse files
my_model/tabs/run_inference.py
CHANGED
@@ -10,7 +10,7 @@ from streamlit.delta_generator import DeltaGenerator
|
|
10 |
from PIL import Image
|
11 |
import torch.nn as nn
|
12 |
import pandas as pd
|
13 |
-
from my_model.object_detection import detect_and_draw_objects
|
14 |
from my_model.captioner.image_captioning import get_caption
|
15 |
from my_model.utilities.gen_utilities import free_gpu_resources
|
16 |
from my_model.state_manager import StateManager
|
|
|
10 |
from PIL import Image
|
11 |
import torch.nn as nn
|
12 |
import pandas as pd
|
13 |
+
from my_model.detector.object_detection import detect_and_draw_objects
|
14 |
from my_model.captioner.image_captioning import get_caption
|
15 |
from my_model.utilities.gen_utilities import free_gpu_resources
|
16 |
from my_model.state_manager import StateManager
|