hamdie commited on
Commit
29fb26e
·
verified ·
1 Parent(s): db73f77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
- from transformers import AutoModelForSequenceClassification
3
 
4
- model = AutoModelForSequenceClassification.from_pretrained('..\arabert_pretrained')
5
 
6
  def sentiment_analysis(text):
7
  prediction = model.predict(text)
 
1
  import gradio as gr
2
+ import transformers as tf
3
 
4
+ model = tf.saved_model.load('..\arabert_pretrained')
5
 
6
  def sentiment_analysis(text):
7
  prediction = model.predict(text)