Merge branch 'main' of https://huggingface.co/spaces/enio/sentiment
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
import tensorflow as tf
|
3 |
|
4 |
-
reloaded_model = tf.saved_model.load("./")
|
5 |
|
6 |
def sentiment(text):
|
7 |
reloaded_results = tf.sigmoid(reloaded_model(tf.constant([text])))
|
|
|
1 |
import gradio as gr
|
2 |
import tensorflow as tf
|
3 |
|
4 |
+
reloaded_model = tf.saved_model.load("./imdb_bert")
|
5 |
|
6 |
def sentiment(text):
|
7 |
reloaded_results = tf.sigmoid(reloaded_model(tf.constant([text])))
|