nsy commited on
Commit
8ca3f79
·
1 Parent(s): 52bad0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ def code_summary(inputs,sum_prompt,num_beam, sec):
84
  generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
85
  output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
86
  elif sec == "Java":
87
- input_ids = tokenizer.encode(inputs + sum_prompt, return_tensors='pt'
88
  generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
89
  output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
90
  s3.pop(output)
 
84
  generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
85
  output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
86
  elif sec == "Java":
87
+ input_ids = tokenizer.encode(inputs + sum_prompt, return_tensors='pt')
88
  generated_ids = sum_model.generate(input_ids, max_length=25,do_sample=False,num_beams=num_beam)
89
  output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
90
  s3.pop(output)