app.py

import gradio as gr
import spaces
import argparse


parser = argparse.ArgumentParser()
# parser.add_argument("--device", type=str, default="cuda:0")
# parser.add_argument("--ckpt_path", type=str, default="./salmonn_7b_v0.pth")
# parser.add_argument("--whisper_path", type=str, default="./whisper_large_v2")
# parser.add_argument("--beats_path", type=str, default="./beats/BEATs_iter3_plus_AS2M_finetuned_on_AS2M_cpt2.pt")
# parser.add_argument("--vicuna_path", type=str, default="./vicuna-7b-v1.5")
# parser.add_argument("--low_resource", action='store_true', default=False)
parser.add_argument("--port", default=9527)

args = parser.parse_args()
args.low_resource = True

title = """<h1 style="text-align: center;">Product description generator</h1>"""
css = """
div#col-container {
    margin: 0 auto;
    max-width: 840px;
}
"""

with gr.Blocks(css=css) as demo:
    with gr.Column(elem_id="col-container"):
        gr.HTML(title)
        gr.Image(type="pil")
        gr.Button(value="Upload")


demo.launch()