akhaliq HF Staff commited on
Commit
8915b24
·
1 Parent(s): bce1fc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -15
app.py CHANGED
@@ -1,18 +1,5 @@
1
- import os
2
- from onnxt5 import GenerativeT5
3
- from onnxt5.api import get_encoder_decoder_tokenizer
4
  import gradio as gr
5
 
6
- os.system("pip install onnxruntime==1.6.0")
7
-
8
-
9
- decoder_sess, encoder_sess, tokenizer = get_encoder_decoder_tokenizer()
10
- generative_t5 = GenerativeT5(encoder_sess, decoder_sess, tokenizer, onnx=True)
11
-
12
- def inference(prompt):
13
- output_text, output_logits = generative_t5(prompt, max_length=100, temperature=0.)
14
- return output_text
15
-
16
  title="T5"
17
- description="T5 is a transformer model which aims to provide great flexibility and provide better semantic understanding through the training of multiple tasks at once."
18
- gr.Interface(inference,"text","text",title=title,description=description).launch(enable_queue=True)
 
 
 
 
1
  import gradio as gr
2
 
 
 
 
 
 
 
 
 
 
 
3
  title="T5"
4
+
5
+ gr.Interface.load("huggingface/t5-base").launch()