mdeniz1's picture
Update app.py
98fcba8
raw
history blame
No virus
451 Bytes
import gradio as gr
from transformers import pipeline
model = pipeline("text-classification", model="mdeniz1/turkish-sentiment-analysis-bert-base-turkish-uncased")
def predict(prompt):
sentiment=model(prompt)
if sentiment[0]['label']=='LABEL_2':
return 'positive'
elif sentiment[0]['label']=='LABEL_1':
return 'neutral'
else:
return 'negative'
gr.Interface(fn=predict, inputs="text", outputs="text").launch()