Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|