Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ import gradio as gr
|
|
5 |
import torch.nn.functional as F
|
6 |
|
7 |
from transformers import AutoTokenizer, GPT2LMHeadModel
|
8 |
-
tokenizer = AutoTokenizer.from_pretrained("supermy/jinyong")
|
9 |
-
model = GPT2LMHeadModel.from_pretrained("supermy/jinyong")
|
10 |
model.eval()
|
11 |
|
12 |
def top_k_top_p_filtering( logits, top_k=0, top_p=0.0, filter_value=-float('Inf') ):
|
|
|
5 |
import torch.nn.functional as F
|
6 |
|
7 |
from transformers import AutoTokenizer, GPT2LMHeadModel
|
8 |
+
tokenizer = AutoTokenizer.from_pretrained("supermy/jinyong-gpt2")
|
9 |
+
model = GPT2LMHeadModel.from_pretrained("supermy/jinyong-gpt2")
|
10 |
model.eval()
|
11 |
|
12 |
def top_k_top_p_filtering( logits, top_k=0, top_p=0.0, filter_value=-float('Inf') ):
|