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():
|