Jezia commited on
Commit
39f1739
1 Parent(s): 3e2cf8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ description = "Gradio Demo for a miniature with GPT. To use it, simply add your
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("aditi2222/automatic_title_generation")
12
 
13
- model = AutoModelForSeq2SeqLM.from_pretrained("aditi2222/automatic_title_generation")
14
 
15
 
16
  def tokenize_data(text):
 
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("aditi2222/automatic_title_generation")
12
 
13
+ model = from_pretrained_keras("keras-io/text-generation-miniature-gpt")
14
 
15
 
16
  def tokenize_data(text):