supermy commited on
Commit
cd80a36
1 Parent(s): 5bb0a8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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') ):