File size: 883 Bytes
4be87b6
93898b2
 
4be87b6
 
 
 
 
 
 
 
 
 
 
93898b2
 
 
 
 
 
 
 
 
 
4be87b6
93898b2
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
import gradio as gr

model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")

def index(prompt):
    inputs = tokenizer(prompt, return_tensors="pt")
    outputs = model.generate(**inputs)

    print(tokenizer.batch_decode(outputs, skip_special_tokens=True))

    return tokenizer.batch_decode(outputs, skip_special_tokens=True)


inputs_image_url = [
    gr.Textbox(type="text", label="Topic Name"),
]

outputs_result_dict = [
    gr.Textbox(type="text", label="Result"),
]

interface_image_url = gr.Interface(
    fn=index,
    inputs=inputs_image_url,
    outputs=outputs_result_dict,
    title="Text Generation",
    cache_examples=False,
)

gr.TabbedInterface(
    [interface_image_url],
    tab_names=['Some inference']
).queue().launch()