Spaces:

Jaehan
/

Text2Text-Text-Summarization

Runtime error

Jaehan commited on Jun 25, 2023

Commit

5d8d571

1 Parent(s): 8dded45

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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