danielolusipe
commited on
Commit
•
af8f9f8
1
Parent(s):
06c6f22
bug_fixes
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import tensorflow as tf
|
|
|
3 |
|
4 |
def split_char(text):
|
5 |
return " ".join(list(text))
|
@@ -56,13 +57,13 @@ def make_predictions(Input):
|
|
56 |
|
57 |
# Loading in model and getting a summary of loaded model
|
58 |
#skimlit_universal_sentence_encoder_model=tf.keras.models.load_model("/content/drive/MyDrive/skimlit_models/Universal_sentence_encoder_Tribrid_embedding_model")
|
59 |
-
|
60 |
|
61 |
# Making prediction with loaded model on sample abstract
|
62 |
-
abstract_pred_probs=
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
|
67 |
# Turning model's prediction into labels
|
68 |
abstract_preds=tf.argmax(abstract_pred_probs,axis=1)
|
|
|
1 |
import gradio as gr
|
2 |
import tensorflow as tf
|
3 |
+
from tensorflow.keras.models import load_model
|
4 |
|
5 |
def split_char(text):
|
6 |
return " ".join(list(text))
|
|
|
57 |
|
58 |
# Loading in model and getting a summary of loaded model
|
59 |
#skimlit_universal_sentence_encoder_model=tf.keras.models.load_model("/content/drive/MyDrive/skimlit_models/Universal_sentence_encoder_Tribrid_embedding_model")
|
60 |
+
skimlit_model=load_model("Universal_sentence_encoder_Tribrid_embedding_model")
|
61 |
|
62 |
# Making prediction with loaded model on sample abstract
|
63 |
+
abstract_pred_probs=skimlit_model.predict(x=(abstract_line_number_one_hot,
|
64 |
+
abstract_total_lines_one_hot,
|
65 |
+
tf.constant(abstract_sentences),
|
66 |
+
tf.constant(abstract_char)))
|
67 |
|
68 |
# Turning model's prediction into labels
|
69 |
abstract_preds=tf.argmax(abstract_pred_probs,axis=1)
|