Rajagopal commited on
Commit
adc4c89
·
1 Parent(s): 4a2cb3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,8 +22,7 @@ def query(payload):
22
  return json.loads(response.content.decode("utf-8"))
23
 
24
 
25
- def inference(input_sentence, max_length, sample_or_greedy, seed=42):
26
- sample_or_greedy = "Greedy"
27
  if sample_or_greedy == "Sample":
28
  parameters = {
29
  "max_new_tokens": max_length,
 
22
  return json.loads(response.content.decode("utf-8"))
23
 
24
 
25
+ def inference(input_sentence, max_length, sample_or_greedy="Greedy", seed=42):
 
26
  if sample_or_greedy == "Sample":
27
  parameters = {
28
  "max_new_tokens": max_length,