alex6095 commited on
Commit
06b90ca
1 Parent(s): a7d7bb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,8 +1,10 @@
 
 
 
 
1
  from transformers import PreTrainedTokenizerFast
2
  from tokenizers import SentencePieceBPETokenizer
3
  from transformers import BartForConditionalGeneration
4
- import streamlit as st
5
- import torch
6
 
7
  def tokenizer():
8
  tokenizer = PreTrainedTokenizerFast.from_pretrained('gogamza/kobart-summarization')
 
1
+ import torch
2
+ import re
3
+ import streamlit as st
4
+
5
  from transformers import PreTrainedTokenizerFast
6
  from tokenizers import SentencePieceBPETokenizer
7
  from transformers import BartForConditionalGeneration
 
 
8
 
9
  def tokenizer():
10
  tokenizer = PreTrainedTokenizerFast.from_pretrained('gogamza/kobart-summarization')