Spaces:

deep-learning-analytics
/

Title_Generation

Runtime error

deep-learning-analytics commited on Jan 23, 2022

Commit

4b98af4

1 Parent(s): c79c43d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,9 +4,9 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-tokenizer = AutoTokenizer.from_pretrained("aditi2222/automatic_title_generation")
-model = AutoModelForSeq2SeqLM.from_pretrained("aditi2222/automatic_title_generation")
 def tokenize_data(text):

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("deep-learning-analytics/automatic-title-generation")
+model = AutoModelForSeq2SeqLM.from_pretrained("deep-learning-analytics/automatic-title-generation")
 def tokenize_data(text):