mdeniz1's picture
Update app.py
fe6772f
raw
history blame
No virus
417 Bytes
import gradio as gr
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()