oucgc1996 commited on
Commit
002ef29
·
verified ·
1 Parent(s): 41d85ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,6 +7,7 @@ import torch.nn as nn
7
  from collections import OrderedDict
8
  import warnings
9
  import gradio as gr
 
10
 
11
  warnings.filterwarnings('ignore')
12
  set_seed(4)
@@ -63,7 +64,7 @@ def ACE(file):
63
  seq_all = []
64
  output_all = []
65
  probability_all = []
66
- for seq in all:
67
  test_data = tokenizer(seq, max_length=max_len, padding="max_length",truncation=True, return_tensors='pt')
68
  out_probability = []
69
  with torch.no_grad():
 
7
  from collections import OrderedDict
8
  import warnings
9
  import gradio as gr
10
+ from tqdm import tqdm
11
 
12
  warnings.filterwarnings('ignore')
13
  set_seed(4)
 
64
  seq_all = []
65
  output_all = []
66
  probability_all = []
67
+ for seq in tqdm(all):
68
  test_data = tokenizer(seq, max_length=max_len, padding="max_length",truncation=True, return_tensors='pt')
69
  out_probability = []
70
  with torch.no_grad():