Update app.py
Browse files
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])
|