prateekbh commited on
Commit
563f98d
1 Parent(s): bbb7d4b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import spaces
3
+ import argparse
4
+
5
+
6
+ parser = argparse.ArgumentParser()
7
+ # parser.add_argument("--device", type=str, default="cuda:0")
8
+ # parser.add_argument("--ckpt_path", type=str, default="./salmonn_7b_v0.pth")
9
+ # parser.add_argument("--whisper_path", type=str, default="./whisper_large_v2")
10
+ # parser.add_argument("--beats_path", type=str, default="./beats/BEATs_iter3_plus_AS2M_finetuned_on_AS2M_cpt2.pt")
11
+ # parser.add_argument("--vicuna_path", type=str, default="./vicuna-7b-v1.5")
12
+ # parser.add_argument("--low_resource", action='store_true', default=False)
13
+ parser.add_argument("--port", default=9527)
14
+
15
+ args = parser.parse_args()
16
+ args.low_resource = True
17
+
18
+ title = """<h1 style="text-align: center;">SALMONN: Speech Audio Language Music Open Neural Network</h1>"""
19
+ css = """
20
+ div#col-container {
21
+ margin: 0 auto;
22
+ max-width: 840px;
23
+ }
24
+ """
25
+
26
+ with gr.Blocks(css=css) as demo:
27
+ with gr.Column(elem_id="col-container"):
28
+ gr.HTML(title)
29
+
30
+
31
+ demo.launch()
32
+
33
+