aavetis commited on
Commit
a3339ce
1 Parent(s): ec7b89f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,3 +1,9 @@
1
  import gradio as gr
2
-
3
- gr.Interface.load("models/microsoft/biogpt").launch()
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from transformers import pipeline, set_seed
3
+ from transformers import BioGptTokenizer, BioGptForCausalLM
4
+ model = BioGptForCausalLM.from_pretrained("microsoft/biogpt")
5
+ tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
6
+ generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
7
+ set_seed(42)
8
+ generator("Tirzepatide is a new drug used to treat", max_length=20, num_return_sequences=5, do_sample=True)
9
+ # gr.Interface.load("models/microsoft/biogpt").launch()