vitiugin commited on
Commit
3d015fd
·
verified ·
1 Parent(s): 37a415b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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', use_auth_token=True)
16
  feedback_labels = ['Request', 'Thanks', 'Question', 'Opinion', 'Concern']
17
 
18
- area_class_model = tf.keras.models.load_model('models/loop_area', use_auth_token=True)
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