Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ mdl = AutoModelWithLMHead.from_pretrained("deep-learning-analytics/wikihow-t5-sm
|
|
6 |
|
7 |
def text2text_summary(para):
|
8 |
initial_txt = para.strip().replace("\n","")
|
9 |
-
tkn_text = text2text_tkn.encode(
|
10 |
|
11 |
tkn_ids = mdl.generate(
|
12 |
tkn_text,
|
|
|
6 |
|
7 |
def text2text_summary(para):
|
8 |
initial_txt = para.strip().replace("\n","")
|
9 |
+
tkn_text = text2text_tkn.encode(initial_txt, return_tensors="pt")
|
10 |
|
11 |
tkn_ids = mdl.generate(
|
12 |
tkn_text,
|