Spaces:
Runtime error
Runtime error
cspocketindia
commited on
Commit
•
760ecf9
1
Parent(s):
ce6c474
Update field name
Browse files- gradio_app.py +3 -3
gradio_app.py
CHANGED
@@ -7,9 +7,9 @@ model = BERTClassifier(model_name='jeevavijay10/nlp-goemotions-bert')
|
|
7 |
|
8 |
classes = Utility().read_emotion_list()
|
9 |
|
10 |
-
def predict(
|
11 |
-
print(
|
12 |
-
predictions = model.evaluate([
|
13 |
print(f"Predictions: {predictions}")
|
14 |
return classes[predictions[0]]
|
15 |
|
|
|
7 |
|
8 |
classes = Utility().read_emotion_list()
|
9 |
|
10 |
+
def predict(sentence):
|
11 |
+
print(sentence)
|
12 |
+
predictions = model.evaluate([sentence])
|
13 |
print(f"Predictions: {predictions}")
|
14 |
return classes[predictions[0]]
|
15 |
|