pluscoloring / app.py
tomyswall's picture
Update app.py
3d3b40e
raw
history blame contribute delete
523 Bytes
import gradio as gr
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
model = AutoModelForSequenceClassification.from_pretrained("MrHup/coloring-book")
tokenizer = AutoTokenizer.from_pretrained("MrHup/coloring-book")
def greet(text):
pipe = pipeline("sentiment-analysis", tokenizer=tokenizer, model=model)
return pipe(text)[0]['label']
iface = gr.Interface(fn=greet, inputs=gr.inputs.Textbox(placeholder="Lütfen Cümle Giriniz...", lines=5), outputs="text")
iface.launch()