Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,17 +5,14 @@ import tensorflow as tf
|
|
| 5 |
|
| 6 |
from sentence_transformers import SentenceTransformer
|
| 7 |
from huggingface_hub import from_pretrained_keras
|
| 8 |
-
from huggingface_hub
|
| 9 |
-
|
| 10 |
-
huggingface_hub.login(HUGGING_FACE_HUB_TOKEN)
|
| 11 |
|
| 12 |
emb_model = SentenceTransformer('distiluse-base-multilingual-cased-v2')
|
| 13 |
EMB_DIM = 512
|
| 14 |
|
| 15 |
-
feedback_class_model = from_pretrained_keras('vitiugin/loop_feedback',
|
| 16 |
feedback_labels = ['Request', 'Thanks', 'Question', 'Opinion', 'Concern']
|
| 17 |
|
| 18 |
-
area_class_model = tf.keras.models.load_model('
|
| 19 |
area_labels = ['cross-cutting', 'education', 'food security', 'governance', 'health', 'protection', 'shelter', 'wash']
|
| 20 |
|
| 21 |
|
|
|
|
| 5 |
|
| 6 |
from sentence_transformers import SentenceTransformer
|
| 7 |
from huggingface_hub import from_pretrained_keras
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
emb_model = SentenceTransformer('distiluse-base-multilingual-cased-v2')
|
| 10 |
EMB_DIM = 512
|
| 11 |
|
| 12 |
+
feedback_class_model = from_pretrained_keras('vitiugin/loop_feedback', token=HUGGING_FACE_HUB_TOKEN)
|
| 13 |
feedback_labels = ['Request', 'Thanks', 'Question', 'Opinion', 'Concern']
|
| 14 |
|
| 15 |
+
area_class_model = tf.keras.models.load_model('vitiugin/loop_area', token=HUGGING_FACE_HUB_TOKEN)
|
| 16 |
area_labels = ['cross-cutting', 'education', 'food security', 'governance', 'health', 'protection', 'shelter', 'wash']
|
| 17 |
|
| 18 |
|