Hugging Face's logo

Spaces:
gradio
/
chatbot
Runtime error

App Files Community
Abubakar Abid commited on
Commit
671a041
·
1 Parent(s): 5bde144

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import GPTNeoForCausalLM, GPT2Tokenizer
2
- model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-125M")
3
- tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
4
 
5
  prompt = """This is a discussion between a person and an entrepreneur.
6
 
 
1
  from transformers import GPTNeoForCausalLM, GPT2Tokenizer
2
+ model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
3
+ tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
4
 
5
  prompt = """This is a discussion between a person and an entrepreneur.
6