Update app.py
Browse files
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():
|