Spaces:
Sleeping
Sleeping
DongfuJiang
commited on
Commit
•
7ad9a04
1
Parent(s):
12cfd21
update
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import os
|
2 |
-
os.environ["HF_HOME"] = "/data/.huggingface"
|
3 |
import gradio as gr
|
4 |
import sys
|
5 |
from datasets import load_dataset
|
@@ -24,7 +24,7 @@ for i, ex in enumerate(SHUFFLED_EXAMPLES_DATASET.take(100)):
|
|
24 |
# continue
|
25 |
EXAMPLES.append([ex[field] for field in fields])
|
26 |
|
27 |
-
scorer = TIGERScorer("TIGER-Lab/TIGERScore-
|
28 |
|
29 |
def submit_fn(input_context, generation_instruction, hypo_output, max_new_tokens=512, temperature=0.7, top_p=1.0):
|
30 |
return scorer.score(
|
@@ -35,7 +35,6 @@ def submit_fn(input_context, generation_instruction, hypo_output, max_new_tokens
|
|
35 |
temperature=temperature,
|
36 |
top_p=top_p,
|
37 |
)[0]['raw_output'].strip()
|
38 |
-
return "None"
|
39 |
|
40 |
|
41 |
|
|
|
1 |
import os
|
2 |
+
# os.environ["HF_HOME"] = "/data/.huggingface"
|
3 |
import gradio as gr
|
4 |
import sys
|
5 |
from datasets import load_dataset
|
|
|
24 |
# continue
|
25 |
EXAMPLES.append([ex[field] for field in fields])
|
26 |
|
27 |
+
scorer = TIGERScorer("TIGER-Lab/TIGERScore-13B-GGUF", use_llamacpp=True)
|
28 |
|
29 |
def submit_fn(input_context, generation_instruction, hypo_output, max_new_tokens=512, temperature=0.7, top_p=1.0):
|
30 |
return scorer.score(
|
|
|
35 |
temperature=temperature,
|
36 |
top_p=top_p,
|
37 |
)[0]['raw_output'].strip()
|
|
|
38 |
|
39 |
|
40 |
|