inoxt commited on
Commit
983c0d0
1 Parent(s): 0dc2ef6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -5,7 +5,6 @@ def greet(name):
5
  checkpoint = "Salesforce/codegen-350M-mono"
6
  model = AutoModelForCausalLM.from_pretrained(checkpoint)
7
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
8
- return "Hello " + name + "!!"
9
  text = "def hello_world():"
10
  completion = model.generate(**tokenizer(text, return_tensors="pt"))
11
  return tokenizer.decode(completion[0])
5
  checkpoint = "Salesforce/codegen-350M-mono"
6
  model = AutoModelForCausalLM.from_pretrained(checkpoint)
7
  tokenizer = AutoTokenizer.from_pretrained(checkpoint)
 
8
  text = "def hello_world():"
9
  completion = model.generate(**tokenizer(text, return_tensors="pt"))
10
  return tokenizer.decode(completion[0])