Commit
•
3adca98
1
Parent(s):
24dbe3e
Update app.py (#1)
Browse files- Update app.py (1c39bd27a27ed07f7bb4e57e3e6a636208a1d0ff)
Co-authored-by: himanish prakash <himanish23@users.noreply.huggingface.co>
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
| 2 |
import gradio as gr
|
| 3 |
import torch
|
| 4 |
-
|
| 5 |
trained_tokenizer = GPT2Tokenizer.from_pretrained("Kumarkishalaya/GPT-2-next-word-prediction")
|
| 6 |
trained_model = GPT2LMHeadModel.from_pretrained("Kumarkishalaya/GPT-2-next-word-prediction")
|
| 7 |
untrained_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|
|
|
|
| 1 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
|
| 2 |
import gradio as gr
|
| 3 |
import torch
|
| 4 |
+
## calling the gpt model
|
| 5 |
trained_tokenizer = GPT2Tokenizer.from_pretrained("Kumarkishalaya/GPT-2-next-word-prediction")
|
| 6 |
trained_model = GPT2LMHeadModel.from_pretrained("Kumarkishalaya/GPT-2-next-word-prediction")
|
| 7 |
untrained_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|