JakeTurner616
commited on
Update app.py
Browse files
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("
|
6 |
-
model = GPTNeoModel.from_pretrained("
|
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:
|