wli3221134 commited on
Commit
0a63b23
·
verified ·
1 Parent(s): c9aca5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -1,9 +1,18 @@
1
- import gradio as gr
2
- import os
3
  import torch
 
 
 
 
 
 
 
4
  from model import Wav2Vec2BERT_Llama # 自定义模型模块
5
  import dataset # 自定义数据集模块
6
 
 
 
 
 
7
  # 初始化设备
8
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
9
 
 
 
 
1
  import torch
2
+
3
+ import spaces
4
+ import os
5
+ import gradio as gr
6
+ from huggingface_hub import HfApi
7
+ from gradio_client.exceptions import AuthenticationError
8
+
9
  from model import Wav2Vec2BERT_Llama # 自定义模型模块
10
  import dataset # 自定义数据集模块
11
 
12
+ @spaces.GPU
13
+ def dummy(): # just a dummy
14
+ pass
15
+
16
  # 初始化设备
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18