Adam173 commited on
Commit
6a4162e
·
1 Parent(s): 783a028

update repo name

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,8 +1,7 @@
1
  from transformers import pipeline, AutoModelWithLMHead, AutoTokenizer
2
  import gradio as gr
3
 
4
- model = AutoModelWithLMHead.from_pretrained("Aethemba/seinfeld-dialogue-model")
5
- # tokenizer = AutoTokenizer.from_pretrained("Aethemba/seinfeld-dialogu")
6
 
7
  seinfeld = pipeline('text-generation', model=model, tokenizer='gpt2', max_length=1000)
8
 
 
1
  from transformers import pipeline, AutoModelWithLMHead, AutoTokenizer
2
  import gradio as gr
3
 
4
+ model = AutoModelWithLMHead.from_pretrained("Adam173/seinfeld-dialogue-model")
 
5
 
6
  seinfeld = pipeline('text-generation', model=model, tokenizer='gpt2', max_length=1000)
7