mrm8488 commited on
Commit
55036a1
β€’
1 Parent(s): ced959c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,11 +4,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed, pipeline
4
  title = "SantaCoder πŸŽ… Swift 🍏 Completion"
5
  description = "This is a subspace to make code generation with [SantaCoder fine-tuned on The Stack Swift](https://huggingface.co/mrm8488/santacoder-finetuned-the-stack-swift)"
6
  EXAMPLE_0 = "import SwiftUI\n\nstruct ContentView: View {\n var body: some View {"
 
7
 
8
 
9
  CKPT = "mrm8488/santacoder-finetuned-the-stack-swift"
10
 
11
- examples = [[EXAMPLE_0, 9, 0.6, 42]]
12
  tokenizer = AutoTokenizer.from_pretrained(CKPT)
13
  model = AutoModelForCausalLM.from_pretrained(CKPT, trust_remote_code=True).to("cuda")
14
 
 
4
  title = "SantaCoder πŸŽ… Swift 🍏 Completion"
5
  description = "This is a subspace to make code generation with [SantaCoder fine-tuned on The Stack Swift](https://huggingface.co/mrm8488/santacoder-finetuned-the-stack-swift)"
6
  EXAMPLE_0 = "import SwiftUI\n\nstruct ContentView: View {\n var body: some View {"
7
+ EXAMPLE_1 = "// Make a naviagtion list with the days of the week\nNavigationView {"
8
 
9
 
10
  CKPT = "mrm8488/santacoder-finetuned-the-stack-swift"
11
 
12
+ examples = [[EXAMPLE_0, 9, 0.6, 42], [EXAMPLE_1, 114, 0.6, 42]]
13
  tokenizer = AutoTokenizer.from_pretrained(CKPT)
14
  model = AutoModelForCausalLM.from_pretrained(CKPT, trust_remote_code=True).to("cuda")
15