hamdie's picture
Update app.py
43de91e verified
raw
history blame
345 Bytes
import gradio as gr
from transformers import AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained('..\arabert_pretrained')
def sentiment_analysis(text):
prediction = model.predict(text)
return prediction
iface = gr.Interface(fn=sentiment_analysis, inputs="text", outputs="text")
iface.launch()