katielink commited on
Commit
a2d1e81
β€’
1 Parent(s): f18cf42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -3,11 +3,9 @@ import gradio as gr
3
  import torch
4
  from transformers import pipeline
5
 
6
- name_list = ['microsoft/biogpt']
7
-
8
  examples = [['COVID-19 is'],['A 65-year-old female patient with a past medical history of']]
9
 
10
- pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt")
11
 
12
  title = "BioGPT Demo"
13
  description = """
@@ -25,7 +23,7 @@ io = gr.Interface(
25
  inference,
26
  gr.Textbox(lines=3),
27
  outputs=[
28
- gr.Textbox(lines=3, label="BioGPT"),
29
  ],
30
  title=title,
31
  description=description,
 
3
  import torch
4
  from transformers import pipeline
5
 
 
 
6
  examples = [['COVID-19 is'],['A 65-year-old female patient with a past medical history of']]
7
 
8
+ pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt-large")
9
 
10
  title = "BioGPT Demo"
11
  description = """
 
23
  inference,
24
  gr.Textbox(lines=3),
25
  outputs=[
26
+ gr.Textbox(lines=3, label="BioGPT-Large"),
27
  ],
28
  title=title,
29
  description=description,