File size: 420 Bytes
68e9609
 
2db5787
 
 
 
d3811c5
8e0ad92
2db5787
68e9609
2db5787
 
68e9609
1
2
3
4
5
6
7
8
9
10
11
12
13
import gradio as gr

from transformers import pipeline

pipe = pipeline("text-generation", model="mistralai/Mistral-7B-v0.1")

#generates an AI description of your character
def describe(names):
    return "Hello " + pipe(names) + "!!"

iface = gr.Interface(fn=describe, inputs=gr.Textbox(label="Your DND character",show_label=True), 
            outputs=gr.Textbox(label="The character",show_label=True))
iface.launch()