Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = "
|
7 |
|
8 |
|
9 |
CKPT = "mrm8488/santacoder-finetuned-the-stack-swift"
|
10 |
|
11 |
-
examples = [[EXAMPLE_0,
|
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 |
|