JakeTurner616 commited on
Commit
8f12d1b
·
verified ·
1 Parent(s): fd809b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import GPTNeoModel, GPT2TokenizerFast
3
 
4
  def generate_text(prompt, max_length, temperature, top_p, repetition_penalty):
5
- tokenizer = GPT2TokenizerFast.from_pretrained("models/JakeTurner616/Adonalsium-gpt-neo-1.3B")
6
- model = GPTNeoModel.from_pretrained("models/JakeTurner616/Adonalsium-gpt-neo-1.3B")
7
 
8
  # Check if tokenizer has a padding token, if not, add one
9
  if tokenizer.pad_token is None:
 
2
  from transformers import GPTNeoModel, GPT2TokenizerFast
3
 
4
  def generate_text(prompt, max_length, temperature, top_p, repetition_penalty):
5
+ tokenizer = GPT2TokenizerFast.from_pretrained("JakeTurner616/Adonalsium-gpt-neo-1.3B")
6
+ model = GPTNeoModel.from_pretrained("JakeTurner616/Adonalsium-gpt-neo-1.3B")
7
 
8
  # Check if tokenizer has a padding token, if not, add one
9
  if tokenizer.pad_token is None: