ashutoshzade commited on
Commit
25fde0a
·
verified ·
1 Parent(s): 7a929c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
8
  def get_recommedation(prompt):
9
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
10
  set_seed(42)
11
- output = generator(prompt, max_length=20, num_return_sequences=5, do_sample=True)
12
  return output
13
 
14
  iface = gr.Interface(fn=get_recommedation, inputs="text", outputs="text")
 
8
  def get_recommedation(prompt):
9
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
10
  set_seed(42)
11
+ output = generator(prompt, max_length=200, num_return_sequences=5, do_sample=True)
12
  return output
13
 
14
  iface = gr.Interface(fn=get_recommedation, inputs="text", outputs="text")