Yashnik commited on
Commit
75a4dce
·
verified ·
1 Parent(s): eb89b8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from getpass import getpass
5
 
6
  model = os.getenv('bigcode/starcoder')
7
 
8
- chatbot = pipeline(model=model)
9
 
10
  message_list = []
11
  response_list = []
 
5
 
6
  model = os.getenv('bigcode/starcoder')
7
 
8
+ chatbot = pipeline(task="text-generation")
9
 
10
  message_list = []
11
  response_list = []