kadirnar commited on
Commit
2dee8e6
1 Parent(s): 959c711

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def biogpt(
37
 
38
  en_prompt = translate_to_english(prompt, lang_model_id, base_lang)
39
  generator = pipeline("text-generation", model=biogpt_model_id, device="cuda:0")
40
- output = generator(en_prompt, model=biogpt_model_id, max_length=max_length, num_return_sequences=num_return_sequences, do_sample=True)
41
  output_dict = {}
42
  for i in range(num_return_sequences):
43
  output_dict[str(i+1)] = output[i]['generated_text']
 
37
 
38
  en_prompt = translate_to_english(prompt, lang_model_id, base_lang)
39
  generator = pipeline("text-generation", model=biogpt_model_id, device="cuda:0")
40
+ output = generator(en_prompt, max_length=max_length, num_return_sequences=num_return_sequences, do_sample=True)
41
  output_dict = {}
42
  for i in range(num_return_sequences):
43
  output_dict[str(i+1)] = output[i]['generated_text']