mrm8488 commited on
Commit
a11e75d
Β·
1 Parent(s): acedb91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,12 +3,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed, pipeline
3
 
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 = """// Swift "Hello, World!" Program"""
7
 
8
 
9
  CKPT = "mrm8488/santacoder-finetuned-the-stack-swift"
10
 
11
- examples = [[EXAMPLE_0, 8, 0.6, 42]]
12
  tokenizer = AutoTokenizer.from_pretrained(CKPT)
13
  model = AutoModelForCausalLM.from_pretrained(CKPT, trust_remote_code=True).to("cuda")
14
 
 
3
 
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 {\tvar 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