dolphin / app.py
nroggendorff's picture
Update app.py
3c2a2e2 verified
raw
history blame
323 Bytes
from transformers import pipeline as pl
import gradio as gr
oracle = pl("text-classification")
def pipe(text: str):
return oracle(text)
def main():
interface = gr.Interface(pipe, gr.Textbox(label="Prompt"), gr.Textbox(label="Response"), title="Text Classifier")
if __name__ == "__main__":
interface.launch()