Hugging Face's logo

Spaces:
flash64
/
biogpt-testing
Runtime error

App Files Community
1
flash64 commited on
Commit
a304887
•
1 Parent(s): e45d82f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def biogpt(
36
  ):
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):
 
36
  ):
37
 
38
  en_prompt = translate_to_english(prompt, lang_model_id, base_lang)
39
+ generator = pipeline("text-generation", model=biogpt_model_id, device="cpu")
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):