Spaces:

Jaehan
/

Text2Text-Text-Summarization

Runtime error

Jaehan commited on Jun 25, 2023

Commit

ce377c6

1 Parent(s): 5d8d571

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
-from transformers import AutoTokenizer, AutoModelWithLMHead
 import gradio as gr
 model_name = "deep-learning-analytics/wikihow-t5-small"
-model = AutoModelWithLMHead.from_pretrained(model_name)
 text2text_tokenizer = AutoTokenizer.from_pretrained(model_name)
 def text2text(paragraph):

+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
 model_name = "deep-learning-analytics/wikihow-t5-small"
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 text2text_tokenizer = AutoTokenizer.from_pretrained(model_name)
 def text2text(paragraph):