katielink commited on
Commit
4dcb436
·
1 Parent(s): d6da92a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ examples = [['COVID-19 is'],['A 65-year-old female patient with a past medical h
12
  #print(f"Is CUDA available: {torch.cuda.is_available()}")
13
  #print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
14
 
15
- pipe_biogpt = pipeline("text2text-generation", model="microsoft/biogpt")
16
- pipe_biomedlm = pipeline("text2text-generation", model="stanford-crfm/BioMedLM")
17
 
18
  title = "Compare generative biomedical LLMs!"
19
  description = "This demo compares [BioGPT](https://huggingface.co/microsoft/biogpt) and [BioMedLM](https://huggingface.co/stanford-crfm/BioMedLM)."
 
12
  #print(f"Is CUDA available: {torch.cuda.is_available()}")
13
  #print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
14
 
15
+ pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt")
16
+ pipe_biomedlm = pipeline("text-generation", model="stanford-crfm/BioMedLM")
17
 
18
  title = "Compare generative biomedical LLMs!"
19
  description = "This demo compares [BioGPT](https://huggingface.co/microsoft/biogpt) and [BioMedLM](https://huggingface.co/stanford-crfm/BioMedLM)."