Hugging Face's logo

Spaces:
gradio
/
chatbot
Running

App Files Community
Abubakar Abid commited on
Commit
5e213a4
•
1 Parent(s): c7e71cc

Update app.py

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