katielink commited on
Commit
bb83af2
1 Parent(s): 65debe0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ examples = [['COVID-19 is'],['A 65-year-old female patient with a past medical h
11
  print(f"Is CUDA available: {torch.cuda.is_available()}")
12
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
13
 
14
- pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt")
15
  pipe_biomedlm = pipeline("text-generation", model="stanford-crfm/BioMedLM", device="cuda:0", model_kwargs={"torch_dtype":torch.bfloat16})
16
  pipe_galactica = pipeline("text-generation", model="facebook/galactica-1.3b", device="cuda:0", model_kwargs={"torch_dtype":torch.bfloat16})
17
 
@@ -34,7 +34,7 @@ io = gr.Interface(
34
  outputs=[
35
  gr.Textbox(lines=3, label="BioGPT"),
36
  gr.Textbox(lines=3, label="BioMedLM (fka PubmedGPT)"),
37
- gr.Textbox(lines=3, label="Galactica"),
38
  ],
39
  title=title,
40
  description=description,
 
11
  print(f"Is CUDA available: {torch.cuda.is_available()}")
12
  print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
13
 
14
+ pipe_biogpt = pipeline("text-generation", model="microsoft/BioGPT-Large", device="cuda:0", model_kwargs={"torch_dtype":torch.bfloat16})
15
  pipe_biomedlm = pipeline("text-generation", model="stanford-crfm/BioMedLM", device="cuda:0", model_kwargs={"torch_dtype":torch.bfloat16})
16
  pipe_galactica = pipeline("text-generation", model="facebook/galactica-1.3b", device="cuda:0", model_kwargs={"torch_dtype":torch.bfloat16})
17
 
 
34
  outputs=[
35
  gr.Textbox(lines=3, label="BioGPT"),
36
  gr.Textbox(lines=3, label="BioMedLM (fka PubmedGPT)"),
37
+ gr.Textbox(lines=3, label="Galactica 1.3B"),
38
  ],
39
  title=title,
40
  description=description,