melonking/ScienceModel

1import gradio as gr2from transformers import AutoTokenizer, AutoModelForCausalLM3from transformers import pipeline, set_seed4def vicuna(input):5 6    generator = pipeline('text-generation', model='gpt2')7    set_seed(42)8    generator(input, max_length=30, num_return_sequences=5)9 10 11 12def greet(name):13    return "Hello " + name + "!!"14 15iface = gr.Interface(fn=vicuna, inputs="text", outputs="text")16iface.launch()