kriton commited on
Commit
68558f3
·
1 Parent(s): fb11063

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -1,3 +1,13 @@
 
 
 
 
 
 
 
 
 
 
1
  def genarate_summary(article):
2
  inputs = tokenizer(
3
  'summarize: ' + article,
 
1
+ from transformers import AutoTokenizer
2
+ from transformers import AutoModelForSeq2SeqLM
3
+
4
+ tokenizer = AutoTokenizer.from_pretrained("sdi/greek-text-summarization")
5
+ model = AutoModelForSeq2SeqLM.from_pretrained("sdi/greek-text-summarization")
6
+
7
+ from transformers import pipeline
8
+
9
+ summarizer = pipeline("summarization", model=model)
10
+
11
  def genarate_summary(article):
12
  inputs = tokenizer(
13
  'summarize: ' + article,