Spaces:

i3-lab
/

i3-GPT

Sleeping

FlameF0X commited on 17 days ago

Commit

024e1b5

verified ·

1 Parent(s): f8636f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 from transformers import GPT2LMHeadModel, GPT2TokenizerFast
 # Replace with your HF username and repo name
-MODEL_REPO = "i3-lab/i3-GPT"
 # Load model and tokenizer
 tokenizer = GPT2TokenizerFast.from_pretrained(MODEL_REPO)

 from transformers import GPT2LMHeadModel, GPT2TokenizerFast
 # Replace with your HF username and repo name
+MODEL_REPO = "i3-lab/i3-GPT2"
 # Load model and tokenizer
 tokenizer = GPT2TokenizerFast.from_pretrained(MODEL_REPO)