Spaces:
Runtime error
Runtime error
Update src/demo.py
Browse files- src/demo.py +4 -4
src/demo.py
CHANGED
|
@@ -14,10 +14,10 @@ ground_truth = ""
|
|
| 14 |
TOKEN = os.environ.get("HF_TOKEN", None)
|
| 15 |
|
| 16 |
type2dataset = {
|
| 17 |
-
"re2text-easy": load_dataset('3B-Group/ConvRe', "en-re2text", token=
|
| 18 |
-
"re2text-hard": load_dataset('3B-Group/ConvRe', "en-re2text", token=
|
| 19 |
-
"text2re-easy": load_dataset('3B-Group/ConvRe', "en-text2re", token=
|
| 20 |
-
"text2re-hard": load_dataset('3B-Group/ConvRe', "en-text2re", token=
|
| 21 |
}
|
| 22 |
|
| 23 |
model_id = "meta-llama/Llama-2-7b-chat-hf"
|
|
|
|
| 14 |
TOKEN = os.environ.get("HF_TOKEN", None)
|
| 15 |
|
| 16 |
type2dataset = {
|
| 17 |
+
"re2text-easy": load_dataset('3B-Group/ConvRe', "en-re2text", token="", split="prompt1"),
|
| 18 |
+
"re2text-hard": load_dataset('3B-Group/ConvRe', "en-re2text", token="", split="prompt4"),
|
| 19 |
+
"text2re-easy": load_dataset('3B-Group/ConvRe', "en-text2re", token="", split="prompt1"),
|
| 20 |
+
"text2re-hard": load_dataset('3B-Group/ConvRe', "en-text2re", token="", split="prompt3")
|
| 21 |
}
|
| 22 |
|
| 23 |
model_id = "meta-llama/Llama-2-7b-chat-hf"
|