fCola commited on
Commit
14f0e2f
·
verified ·
1 Parent(s): ccd20ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -434,8 +434,8 @@ paper_plane_svg = """<svg xmlns="http://www.w3.org/2000/svg" width="20" height="
434
 
435
  # Pipeline loading
436
  #generator = pipeline("text-generation", model="openai-community/gpt2")
437
- tokenizer = AutoTokenizer("openai-community/gpt2")
438
- model = AutoModelForCausalLM("openai-community/gpt2")
439
 
440
  # Mock data function for chatbot
441
  def send_message(message, history):
 
434
 
435
  # Pipeline loading
436
  #generator = pipeline("text-generation", model="openai-community/gpt2")
437
+ tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
438
+ model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
439
 
440
  # Mock data function for chatbot
441
  def send_message(message, history):