DongfuJiang commited on
Commit
7ad9a04
1 Parent(s): 12cfd21
Files changed (1) hide show
  1. app.py +2 -3
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-7B")
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