nuojohnchen commited on
Commit
07e8e35
·
verified ·
1 Parent(s): cd84e2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ DESCRIPTION = '''
20
 
21
  LICENSE = """
22
  <div style="font-family: monospace; white-space: pre; margin-top: 20px; line-height: 1.2;">
23
- @misc{XtraGPT,
24
  title = {JudgeLRM},
25
  url = {https://huggingface.co/nuojohnchen/JudgeLRM-7B},
26
  author = {Nuo Chen, Zhiyuan Hu, Qingyun Zou, Jiaying Wu, Qian Wang, Bryan Hooi, Bingsheng He},
@@ -248,7 +248,7 @@ def generate_and_judge(instruction, model_dropdown_1, custom_model_1, model_drop
248
  return response1, response2, None
249
 
250
  # Function to stream evaluation results after responses are generated
251
- @spaces.GPU(duration=120)
252
  def stream_evaluation(instruction, response1, response2, judge_model_name, temperature=0.1, max_new_tokens=2048):
253
  """Stream evaluation results after responses are generated"""
254
  for result in judge_responses(instruction, response1, response2, judge_model_name, temperature, max_new_tokens):
 
20
 
21
  LICENSE = """
22
  <div style="font-family: monospace; white-space: pre; margin-top: 20px; line-height: 1.2;">
23
+ @misc{JudgeLRM,
24
  title = {JudgeLRM},
25
  url = {https://huggingface.co/nuojohnchen/JudgeLRM-7B},
26
  author = {Nuo Chen, Zhiyuan Hu, Qingyun Zou, Jiaying Wu, Qian Wang, Bryan Hooi, Bingsheng He},
 
248
  return response1, response2, None
249
 
250
  # Function to stream evaluation results after responses are generated
251
+ @spaces.GPU(duration=200)
252
  def stream_evaluation(instruction, response1, response2, judge_model_name, temperature=0.1, max_new_tokens=2048):
253
  """Stream evaluation results after responses are generated"""
254
  for result in judge_responses(instruction, response1, response2, judge_model_name, temperature, max_new_tokens):