File size: 936 Bytes
fc1b9c5 bc7d231 c7c92f9 dfda773 356a130 58e3cb5 63fc765 b491c60 5554139 e9d7d81 1a06525 8cf7678 c183786 a650af8 c6252cf ef4d7f5 c183786 f05f048 c183786 f35e4aa fc1b9c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import streamlit as st
import torch
import bitsandbytes
import accelerate
import scipy
import copy
from PIL import Image
import torch.nn as nn
import pandas as pd
from my_model.object_detection import detect_and_draw_objects
from my_model.captioner.image_captioning import get_caption
from my_model.gen_utilities import free_gpu_resources
from my_model.KBVQA import KBVQA, prepare_kbvqa_model
from my_model.utilities.st_utils import UIManager, StateManager
def main():
ui_manager = UIManager()
selection = ui_manager.display_sidebar()
ui_manager.display_selected_page("Home")
ui_manager.display_selected_page("Dataset Analysis")
ui_manager.display_selected_page("Finetuning and Evaluation Results")
ui_manager.display_selected_page("Run Inference")
ui_manager.display_selected_page("Code")
ui_manager.display_selected_page("More Pages will follow .. ")
if __name__ == "__main__":
main() |