File size: 1,106 Bytes
326115c
a84a5a1
 
719ecfd
 
a84a5a1
 
e5c9ee0
8959fb9
 
acec8bf
326115c
719ecfd
326115c
c5fe4a2
 
8959fb9
3bd20e4
a84a5a1
 
 
 
 
 
72ceb76
8959fb9
c41e6ce
8959fb9
c41e6ce
8959fb9
c41e6ce
f623930
c41e6ce
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
import logging
import os
import sys
import gradio as gr
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates

from app import routes
from utils import app_helpers, session_logger


session_logger.change_logging(logging.DEBUG)

CUSTOM_GRADIO_PATH = "/"
app = FastAPI(title="lisa_app", version="1.0")
app.include_router(routes.router)

FASTAPI_STATIC = os.getenv("FASTAPI_STATIC")
os.makedirs(FASTAPI_STATIC, exist_ok=True)
app.mount("/static", StaticFiles(directory=FASTAPI_STATIC), name="static")
templates = Jinja2Templates(directory="templates")


logging.info(f"sys.argv:{sys.argv}.")
args = app_helpers.parse_args([])
logging.info(f"prepared default arguments:{args}.")
inference_fn = app_helpers.get_inference_model_by_args(args)
logging.info(f"prepared inference_fn function:{inference_fn.__name__}, creating gradio interface...")
io = app_helpers.get_gradio_interface(inference_fn)
logging.info("created gradio interface")
app = gr.mount_gradio_app(app, io, path=CUSTOM_GRADIO_PATH)
logging.info("mounted gradio app within fastapi")