hf-dongpyo commited on
Commit
696d1c3
·
1 Parent(s): d89c47c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import gradio as grad
3
 
4
  codegen_tkn = AutoTokenizer.from_pretrained('Salesforce/codegen-350M-mono')
5
- mdl = AutoModelForCausalLM.from_pretrained('Salesforce/codegen-250M-mono')
6
 
7
  def codegen(intent):
8
  # given input as text which reflects intent of the program.
 
2
  import gradio as grad
3
 
4
  codegen_tkn = AutoTokenizer.from_pretrained('Salesforce/codegen-350M-mono')
5
+ mdl = AutoModelForCausalLM.from_pretrained('Salesforce/codegen-350M-mono')
6
 
7
  def codegen(intent):
8
  # given input as text which reflects intent of the program.