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 |
|