sagu7 commited on
Commit
faa48c9
·
1 Parent(s): 40e75ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,11 +25,11 @@ class OutputResponse(BaseModel):
25
 
26
  @app.post("/evaluate")
27
  def evaluate(input_prompt: InputPrompt):
28
- temperature = random.uniform(0.1, 1.0)
29
  generation_config = GenerationConfig(
30
  temperature=temperature,
31
  top_p=0.75,
32
- num_beams=4,
33
  )
34
  prompt = generate_prompt(input_prompt.instruction, input_prompt.input)
35
  inputs = tokenizer(prompt, return_tensors="pt")
 
25
 
26
  @app.post("/evaluate")
27
  def evaluate(input_prompt: InputPrompt):
28
+ temperature = 0.9
29
  generation_config = GenerationConfig(
30
  temperature=temperature,
31
  top_p=0.75,
32
+ num_beams=1,, do_sample=True
33
  )
34
  prompt = generate_prompt(input_prompt.instruction, input_prompt.input)
35
  inputs = tokenizer(prompt, return_tensors="pt")