Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|
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']
|