Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,10 @@ import torch
|
|
4 |
import gradio as gr
|
5 |
import torch.nn.functional as F
|
6 |
|
7 |
-
from transformers import
|
8 |
-
|
|
|
|
|
9 |
|
10 |
def top_k_top_p_filtering( logits, top_k=0, top_p=0.0, filter_value=-float('Inf') ):
|
11 |
assert logits.dim() == 1
|
@@ -58,9 +60,7 @@ if __name__ == '__main__':
|
|
58 |
# sep_id = tokenizer.sep_token_id
|
59 |
# unk_id = tokenizer.unk_token_id
|
60 |
|
61 |
-
|
62 |
-
model = GPT2LMHeadModel.from_pretrained("supermy/poetry")
|
63 |
-
model.eval()
|
64 |
|
65 |
gr.Interface(
|
66 |
fn=generate,
|
|
|
4 |
import gradio as gr
|
5 |
import torch.nn.functional as F
|
6 |
|
7 |
+
from transformers import BertTokenizer, GPT2LMHeadModel
|
8 |
+
tokenizer = BertTokenizer.from_pretrained("supermy/poetry")
|
9 |
+
model = GPT2LMHeadModel.from_pretrained("supermy/poetry")
|
10 |
+
model.eval()
|
11 |
|
12 |
def top_k_top_p_filtering( logits, top_k=0, top_p=0.0, filter_value=-float('Inf') ):
|
13 |
assert logits.dim() == 1
|
|
|
60 |
# sep_id = tokenizer.sep_token_id
|
61 |
# unk_id = tokenizer.unk_token_id
|
62 |
|
63 |
+
|
|
|
|
|
64 |
|
65 |
gr.Interface(
|
66 |
fn=generate,
|