Spaces:
Runtime error
Runtime error
update repo name
Browse files
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("
|
| 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 |
|